Update app.py

app.py

@@ -253,8 +253,8 @@ def predict(image):
     
     # Process output (adjust based on your model's format)
     # return output  # or post-process the results as needed
-    results = model(image, save=True)
-    annotated_img = Image.open("B111.JPG")
+    results = model(image)
+    annotated_img = results.plot()
     return annotated_img
 
 # Gradio interface