Spaces:
Running
Running
import gradio as gr | |
from transformers import pipeline | |
decade = pipeline(model="tonyassi/fashion-clothing-decade") | |
def greet(img): | |
pred = decade(images=img) | |
res = pred[0]['label'] + ':' + str(pred[0]['score']) + '\n' | |
res += pred[1]['label'] + ':' + str(pred[1]['score']) + '\n' | |
res += pred[2]['label'] + ':' + str(pred[2]['score']) | |
return res | |
iface = gr.Interface(fn=greet, inputs=gr.Image(type="pil"), outputs="text") | |
iface.launch() |