add ckpt
Browse files- app.py +2 -2
- model.ckpt +3 -0
app.py
CHANGED
|
@@ -54,7 +54,7 @@ DEMO_POINTCLOUDS = {
|
|
| 54 |
|
| 55 |
|
| 56 |
def initialize_model(
|
| 57 |
-
checkpoint_path="
|
| 58 |
):
|
| 59 |
"""Initialize the model at startup and store it in the global cache."""
|
| 60 |
global PI3DETR_MODEL, MODEL_STATUS
|
|
@@ -836,7 +836,7 @@ def run_model_with_display(
|
|
| 836 |
|
| 837 |
with gr.Blocks(title="PI3DETR") as demo:
|
| 838 |
gr.Markdown(
|
| 839 |
-
"# 🥧 PI3DETR: 3D
|
| 840 |
"An end-to-end deep learning model for **parametric curve inference** in **3D point clouds** and **meshes**.\n"
|
| 841 |
"Upload a `.xyz`, `.ply`, or `.obj` file to explore curve detection."
|
| 842 |
)
|
|
|
|
| 54 |
|
| 55 |
|
| 56 |
def initialize_model(
|
| 57 |
+
checkpoint_path="model.ckpt", config_path="configs/pi3detr.yaml"
|
| 58 |
):
|
| 59 |
"""Initialize the model at startup and store it in the global cache."""
|
| 60 |
global PI3DETR_MODEL, MODEL_STATUS
|
|
|
|
| 836 |
|
| 837 |
with gr.Blocks(title="PI3DETR") as demo:
|
| 838 |
gr.Markdown(
|
| 839 |
+
"# 🥧 PI3DETR: Detection of Sharp 3D CAD Edges [CPU-PREVIEW]\n"
|
| 840 |
"An end-to-end deep learning model for **parametric curve inference** in **3D point clouds** and **meshes**.\n"
|
| 841 |
"Upload a `.xyz`, `.ply`, or `.obj` file to explore curve detection."
|
| 842 |
)
|
model.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f8c66a61b51a5f3ada6769769a3b104bdd68d20e877c068478218dc1eec9a9fb
|
| 3 |
+
size 392993214
|