Update app.py

app.py

@@ -233,9 +233,10 @@ def predict(image):
         output = model(img_tensor)
     
     # Process output (adjust based on your model's format)
-    return output  # or post-process the results as needed
-    # annotated_img = output.render()[0]
-    # return annotated_img
+    # return output  # or post-process the results as needed
+    results = model(img)
+    annotated_img = results.render()[0]
+    return annotated_img
 
 # Gradio interface
 demo = gr.Interface(