Update app.py

app.py

@@ -43,7 +43,7 @@ def predict(img_name, model=None, device=None, reduce_labels=True):
     my_model.load_checkpoints(checkpoints)
     with torch.no_grad():
         output = my_model(pre_img_data)
-    overlay = visualize_instances_map(pre_img_data,star_label)
+    overlay = visualize_instances_map(pre_img_data,output)
 #cv2.imwrite('prediction.png', cv2.cvtColor(overlay, cv2.COLOR_RGB2BGR))
     return overlay
 gr.Interface(