souranil3d commited on
Commit
7b73b04
1 Parent(s): d1d6e36

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=model
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)