Spaces:
Running
Running
File size: 455 Bytes
3bef070 f0ea29c 3bef070 f0ea29c 3bef070 f0ea29c 3bef070 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
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() |