wasmdashai commited on
Commit
c8fa93b
·
verified ·
1 Parent(s): 3ec9dda

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -7
app.py CHANGED
@@ -1,11 +1,6 @@
1
  import gradio as gr
2
  # Use a pipeline as a high-level helper
3
  from transformers import pipeline
 
4
 
5
- pipe = pipeline("image-classification", model="wasmdashai/vit-face-expression-v1")
6
- def greet(image):
7
- out=pipe(image)
8
- return str(out)
9
-
10
- demo = gr.Interface(fn=greet, inputs="image", outputs="text")
11
- demo.launch()
 
1
  import gradio as gr
2
  # Use a pipeline as a high-level helper
3
  from transformers import pipeline
4
+ import gradio as gr
5
 
6
+ gr.load("models/wasmdashai/vit-face-expression-v1").launch()