anjikum commited on
Commit
16a6690
·
verified ·
1 Parent(s): fbfc508

corrected image in gradio args

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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.inputs.Image(type="pil"), outputs="text")
 
 
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()