Spaces:
Sleeping
Sleeping
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.inputs.Image(shape=(256, 256,
|
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.inputs.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)
|