Update app.py

app.py

@@ -33,12 +33,12 @@ if not torch.cuda.is_available():
 predictor = DefaultPredictor(cfg)
 
 def inference(image):
-    # print(image.height)
+    print(image.height)
 
-    # height = image.height
+    height = image.height
 
-    # img = np.array(image.resize((500, height)))
-    outputs = predictor(image)
+    img = np.array(image.resize((500, height)))
+    outputs = predictor(img)
 
     v = Visualizer(img,scale=1.2)
     out = v.draw_instance_predictions(outputs["instances"].to("cpu"))