Spaces:
Running
Running
change Model position download
Browse files
modules/streamlit_utils.py
CHANGED
@@ -61,8 +61,8 @@ def load_models():
|
|
61 |
model_object = get_faster_rcnn_model(len(object_dict))
|
62 |
model_arrow = get_arrow_model(len(arrow_dict), 2)
|
63 |
|
64 |
-
model_arrow_path = hf_hub_download(repo_id="
|
65 |
-
model_object_path = hf_hub_download(repo_id="
|
66 |
|
67 |
# Define paths to save models
|
68 |
output_arrow = 'model_arrow.pth'
|
|
|
61 |
model_object = get_faster_rcnn_model(len(object_dict))
|
62 |
model_arrow = get_arrow_model(len(arrow_dict), 2)
|
63 |
|
64 |
+
model_arrow_path = hf_hub_download(repo_id="ELCA-SA/BPMN_Detection", filename="model_arrow.pth")
|
65 |
+
model_object_path = hf_hub_download(repo_id="ELCA-SA/BPMN_Detection", filename="model_object.pth")
|
66 |
|
67 |
# Define paths to save models
|
68 |
output_arrow = 'model_arrow.pth'
|