Update app.py

app.py

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