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)
|