corrected image in gradio args
Browse files
app.py
CHANGED
@@ -77,7 +77,9 @@ def predict(image):
|
|
77 |
return LABELS[predicted.item()] # Return the predicted class label
|
78 |
|
79 |
# Set up the Gradio interface
|
80 |
-
interface = gr.Interface(fn=predict, inputs=gr.
|
|
|
|
|
81 |
|
82 |
# Launch the interface
|
83 |
interface.launch()
|
|
|
77 |
return LABELS[predicted.item()] # Return the predicted class label
|
78 |
|
79 |
# Set up the Gradio interface
|
80 |
+
interface = gr.Interface(fn=predict, inputs=gr.Image(type="pil"), outputs="text")
|
81 |
+
|
82 |
+
# interface = gr.Interface(fn=predict, inputs=gr.inputs.Image(type="pil"), outputs="text")
|
83 |
|
84 |
# Launch the interface
|
85 |
interface.launch()
|