shewster commited on
Commit
c91b23c
1 Parent(s): e3a5af5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -15,11 +15,7 @@ def predict_image(img):
15
  return {learn.dls.vocab[i]: float(probs[i]) for i in range(len(learn.dls.vocab))}
16
 
17
  # Create a Gradio interface
18
- interface = gr.Interface(fn=predict_image,
19
- inputs=gr.inputs.Image(type="pil"),
20
- outputs=gr.outputs.Label(num_top_classes=3),
21
- title="Image Classifier",
22
- description="Upload an image to classify.")
23
 
24
  if __name__ == "__main__":
25
  interface.launch()
 
15
  return {learn.dls.vocab[i]: float(probs[i]) for i in range(len(learn.dls.vocab))}
16
 
17
  # Create a Gradio interface
18
+ gr.Interface(fn=classify_image,inputs=gr.Image(type=“pil”),outputs=gr.Label())
 
 
 
 
19
 
20
  if __name__ == "__main__":
21
  interface.launch()