antfraia commited on
Commit
d86e9bb
·
1 Parent(s): 401df8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -12
app.py CHANGED
@@ -1,14 +1,3 @@
1
  import gradio as gr
2
 
3
- # Instead of launching the model directly, save it to a variable
4
- model_interface = gr.load("models/google/vit-base-patch16-224")
5
-
6
- # Customize the interface with the desired template properties
7
- model_interface.title = "Image classification"
8
- model_interface.description = "Upload an image here and classify it based on Google models"
9
- model_interface.theme = "Monochrome"
10
- model_interface.live = True
11
- model_interface.capture_session = True
12
-
13
- # Launch the customized Gradio interface
14
- model_interface.launch()
 
1
  import gradio as gr
2
 
3
+ gr.Interface.load("models/google/vit-base-patch16-224").launch()