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() |