DGurgurov commited on
Commit
ea824f4
1 Parent(s): 4b2d5bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,8 +27,8 @@ def predict(image):
27
  return {f"Class {i}": label for i, label in enumerate(top_5)}
28
 
29
  # Define Gradio interface
30
- image = gr.inputs.Image(type="pil")
31
- label = gr.outputs.Label(num_top_classes=5)
32
 
33
  interface = gr.Interface(
34
  fn=predict,
 
27
  return {f"Class {i}": label for i, label in enumerate(top_5)}
28
 
29
  # Define Gradio interface
30
+ image = gr.components.Image(type="pil")
31
+ label = gr.components.Label(num_top_classes=5)
32
 
33
  interface = gr.Interface(
34
  fn=predict,