Update app.py

app.py

@@ -256,7 +256,7 @@ def predict(image):
     results = model(image)
     print(type(results))
     print(results)
-    annotated_img = results.plot()
+    annotated_img = results[0].plot()
     return annotated_img
 
 # Gradio interface