Spaces:
Running
Running
changed of the drive link for models
Browse files
app.py
CHANGED
@@ -155,8 +155,8 @@ def load_models():
|
|
155 |
model_object = get_faster_rcnn_model(len(object_dict))
|
156 |
model_arrow = get_arrow_model(len(arrow_dict),2)
|
157 |
|
158 |
-
url_arrow = 'https://drive.google.com/uc?id=
|
159 |
-
url_object = 'https://drive.google.com/uc?id=
|
160 |
|
161 |
# Define paths to save models
|
162 |
output_arrow = 'model_arrow.pth'
|
|
|
155 |
model_object = get_faster_rcnn_model(len(object_dict))
|
156 |
model_arrow = get_arrow_model(len(arrow_dict),2)
|
157 |
|
158 |
+
url_arrow = 'https://drive.google.com/uc?id=1vv1X_r_lZ8gnzMAIKxcVEb_T_Qb-NkyA'
|
159 |
+
url_object = 'https://drive.google.com/uc?id=1lmeVBRfTrcrBsq3904m4TgppSA4nqsIi'
|
160 |
|
161 |
# Define paths to save models
|
162 |
output_arrow = 'model_arrow.pth'
|