Update app.py

app.py

@@ -4,17 +4,16 @@ from PIL import Image
 import numpy as np
 
 # Load YOLOv12 model
-model = YOLO("best.pt")  # model in repo
+model = YOLO("best.pt")
 
 def predict(image):
-    # Ensure RGB and convert to NumPy array
-    image = image.convert("RGB")
-    img_array = np.array(image)
-    
+    # Ensure RGB and convert to NumPy
+    img_array = np.array(image.convert("RGB"))
+
     # Run inference
-    results = model.predict(img_array)
-    
-    # Annotate and return
+    results = model.predict(img_array, imgsz=640)  # specify size for safety
+
+    # Annotate image
     annotated_img = results[0].plot()
     return annotated_img