Spaces:
Build error
Build error
Commit
·
7b73b04
1
Parent(s):
d1d6e36
Update app.py
Browse files
app.py
CHANGED
@@ -26,7 +26,7 @@ def predict(
|
|
26 |
weights = "checkpoints_model_final_imagenet_40k_synthetic.pth"
|
27 |
else:
|
28 |
weights = "checkpoints_model_final_imagenet_100k_synthetic.pth"
|
29 |
-
cfg.MODEL.WEIGHTS=
|
30 |
predictor = DefaultPredictor(cfg)
|
31 |
im = cv2.imread(img.name)
|
32 |
output = predictor(im)
|
|
|
26 |
weights = "checkpoints_model_final_imagenet_40k_synthetic.pth"
|
27 |
else:
|
28 |
weights = "checkpoints_model_final_imagenet_100k_synthetic.pth"
|
29 |
+
cfg.MODEL.WEIGHTS=weights
|
30 |
predictor = DefaultPredictor(cfg)
|
31 |
im = cv2.imread(img.name)
|
32 |
output = predictor(im)
|