Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -32,7 +32,7 @@ def predict_image(image):
|
|
| 32 |
return {"prediction": predict_label, "confidence": float(np.max(predictions))}
|
| 33 |
|
| 34 |
# Create the interface
|
| 35 |
-
input_interface = gr.
|
| 36 |
output_interface = gr.outputs.Label(num_top_classes=2)
|
| 37 |
|
| 38 |
iface = gr.Interface(fn=predict_image, inputs=input_interface, outputs=output_interface)
|
|
|
|
| 32 |
return {"prediction": predict_label, "confidence": float(np.max(predictions))}
|
| 33 |
|
| 34 |
# Create the interface
|
| 35 |
+
input_interface = gr.Image(shape=(256, 256), image_mode='RGB')
|
| 36 |
output_interface = gr.outputs.Label(num_top_classes=2)
|
| 37 |
|
| 38 |
iface = gr.Interface(fn=predict_image, inputs=input_interface, outputs=output_interface)
|