Spaces:
Runtime error
Runtime error
siddhantgore
commited on
Commit
•
b0f0ca6
1
Parent(s):
199dc75
Update app.py
Browse files
app.py
CHANGED
@@ -26,7 +26,7 @@ import numpy as np
|
|
26 |
|
27 |
model_path = "model_final.pth"
|
28 |
|
29 |
-
|
30 |
cfg.merge_from_file(model_zoo.get_config_file("COCO-Detection/faster_rcnn_R_50_FPN_1x.yaml")) # load the default configuration
|
31 |
cfg.MODEL.WEIGHTS = 'model_final.pth'
|
32 |
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.8
|
|
|
26 |
|
27 |
model_path = "model_final.pth"
|
28 |
|
29 |
+
cfg=get_cfg()
|
30 |
cfg.merge_from_file(model_zoo.get_config_file("COCO-Detection/faster_rcnn_R_50_FPN_1x.yaml")) # load the default configuration
|
31 |
cfg.MODEL.WEIGHTS = 'model_final.pth'
|
32 |
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.8
|