siddhantgore commited on
Commit
a499985
1 Parent(s): 4b0a60e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -54,7 +54,7 @@ def inference(image):
54
  #img = np.array(image.resize((800, height)))
55
  img = np.array(image)
56
  outputs = predictor(img)
57
- v=Visualizer(im,MetadataCatalog.get(dataset_name),scale=1)
58
  out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
59
  return out.get_image()
60
  if file is None:
 
54
  #img = np.array(image.resize((800, height)))
55
  img = np.array(image)
56
  outputs = predictor(img)
57
+ v=Visualizer(img,MetadataCatalog.get(dataset_name),scale=1)
58
  out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
59
  return out.get_image()
60
  if file is None: