Update app.py

app.py

@@ -233,13 +233,15 @@ 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
+    # return output  # or post-process the results as needed
+    annotated_img = output.render()[0]
+    return annotated_img
 
 # Gradio interface
 demo = gr.Interface(
     fn=predict,
     inputs=gr.Image(type="pil"),  # Accepts image input
-    outputs="json"  # Customize based on your output format
+    outputs="image"  # Customize based on your output format
 )
 
 if __name__ == "__main__":