benjaminStreltzin commited on
Commit
f83a3a3
·
verified ·
1 Parent(s): 203c95d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -20,10 +20,11 @@ def predict(image: Image.Image):
20
  # Define the Gradio interface with updated API
21
  demo = gr.Interface(
22
  fn=predict,
23
- inputs=gr.Image(type="pil"), # Updated for new Gradio API
24
- outputs=[gr.Textbox(), gr.Textbox()], # Updated for new Gradio API
25
- title="Vision Transformer Model", # Title of the Gradio interface
26
- description="Upload an image to classify it using the Vision Transformer model." # Description
 
27
  )
28
 
29
  # Launch the Gradio interface
 
20
  # Define the Gradio interface with updated API
21
  demo = gr.Interface(
22
  fn=predict,
23
+ inputs=gr.Image(type="pil"),
24
+ outputs=[gr.Textbox(), gr.Textbox()],
25
+ title="Vision Transformer Model",
26
+ description="Upload an image to classify it using the Vision Transformer model.",
27
+ load=gr.Video("load_screen.mp4") # Specify the loading video
28
  )
29
 
30
  # Launch the Gradio interface