Update app.py

app.py

@@ -233,9 +233,9 @@ 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
+    # annotated_img = output.render()[0]
+    # return annotated_img
 
 # Gradio interface
 demo = gr.Interface(