Spaces:
Running
Running
Update my_model/object_detection.py
Browse files
my_model/object_detection.py
CHANGED
@@ -61,6 +61,7 @@ class ObjectDetector:
|
|
61 |
|
62 |
try:
|
63 |
model_path = get_model_path('deformable-detr-detic')
|
|
|
64 |
self.processor = AutoImageProcessor.from_pretrained(model_path)
|
65 |
self.model = AutoModelForObjectDetection.from_pretrained(model_path)
|
66 |
except Exception as e:
|
|
|
61 |
|
62 |
try:
|
63 |
model_path = get_model_path('deformable-detr-detic')
|
64 |
+
st.write(model_path)
|
65 |
self.processor = AutoImageProcessor.from_pretrained(model_path)
|
66 |
self.model = AutoModelForObjectDetection.from_pretrained(model_path)
|
67 |
except Exception as e:
|