Update app.py

app.py

@@ -233,7 +233,7 @@ def predict(image):
         output = model(img_tensor)
     
     # Process output (adjust based on your model's format)
-    return output.tolist()  # or post-process the results as needed
+    return output  # or post-process the results as needed
 
 # Gradio interface
 demo = gr.Interface(