DGurgurov commited on
Commit
bcd33a3
1 Parent(s): 103d031

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -18,8 +18,8 @@ def predict(image):
18
  return {f"Class {i}": prob.item() for i, prob in enumerate(probs[0])}
19
 
20
  # Define Gradio interface
21
- image = gr.inputs.Image(type="pil")
22
- label = gr.outputs.Label(num_top_classes=5)
23
 
24
  interface = gr.Interface(
25
  fn=predict,
 
18
  return {f"Class {i}": prob.item() for i, prob in enumerate(probs[0])}
19
 
20
  # Define Gradio interface
21
+ image = gr.components.Image(type="pil")
22
+ label = gr.components.Label(num_top_classes=5)
23
 
24
  interface = gr.Interface(
25
  fn=predict,