BenjiELCA commited on
Commit
91857b0
·
1 Parent(s): 8d48eae

change Model position download

Browse files
Files changed (1) hide show
  1. modules/streamlit_utils.py +2 -2
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="BenjiELCA/BPMN_Detection", filename="model_arrow.pth")
65
- model_object_path = hf_hub_download(repo_id="BenjiELCA/BPMN_Detection", filename="model_object.pth")
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'