face / app.py
wasmdashai's picture
Create app.py
004b486 verified
raw
history blame
313 Bytes
import gradio as gr
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("image-classification", model="wasmdashai/vit-face-expression-v1")
def greet(image):
out=pipe(image)
return str(out)
demo = gr.Interface(fn=greet, inputs="image", outputs="text")
demo.launch()