Update app.py

app.py

@@ -10,6 +10,7 @@ from PIL import Image
 models = ["model_weights/md_v5a.0.0.pt","model_weights/md_v5b.0.0.pt"]
 
 def yolo(image, chosen_model, size):
+    im = image
     g = (size / max(im.size))  # gain
     im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS)  # resize