asalhi85 commited on
Commit
de4171c
·
1 Parent(s): cc5c4a7

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ def inference(image_url, image, min_score):
55
  cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = min_score
56
  predictor = DefaultPredictor(cfg)
57
 
58
- outputs = predictor(im)
59
 
60
  v = Visualizer(image[:,:,::-1], my_metadata, scale=1.2, instance_mode=ColorMode.IMAGE )
61
  out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
 
55
  cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = min_score
56
  predictor = DefaultPredictor(cfg)
57
 
58
+ outputs = predictor(image)
59
 
60
  v = Visualizer(image[:,:,::-1], my_metadata, scale=1.2, instance_mode=ColorMode.IMAGE )
61
  out = v.draw_instance_predictions(outputs["instances"].to("cpu"))