Update app.py

app.py

@@ -15,8 +15,11 @@ def predict(image):
 
     # Perform inference
     predictions = model(img)
-    return predictions.numpy()  # Adjust output processing as needed
+    return predictions.numpy().tolist()  # Adjust output processing as needed
 
 # Set up the Gradio interface
-interface = gr.Interface(fn=predict, inputs=gr.inputs.Image(type="pil"), outputs="label")
+image_input = gr.inputs.Image(shape=(640, 640))
+label_output = gr.outputs.Label(num_top_classes=3)
+
+interface = gr.Interface(fn=predict, inputs=image_input, outputs=label_output)
 interface.launch()