LucyintheSky's picture
Update app.py
7000e84
raw
history blame
280 Bytes
import gradio as gr
from transformers import pipeline
crop_pipe = pipeline("image-classification", model="LucyintheSky/pose-estimation-crop-uncrop")
def classify(img):
return crop_pipe(img)
iface = gr.Interface(fn=classify, inputs=gr.Image(), outputs="text")
iface.launch()