File size: 938 Bytes
3bef070
f0ea29c
3bef070
f0ea29c
3bef070
f0ea29c
 
 
 
 
 
 
738ad39
 
 
 
 
 
 
 
c9dcb6a
738ad39
 
3bef070
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
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,
                     title='Which Decade Are You From?',
                     description="""
by [Tony Assi](https://www.tonyassi.com/)
                     
This space uses the [fashion-clothing-decade image classification model](https://huggingface.co/tonyassi/fashion-clothing-decade). Please ❤️ this Space.

I build custom AI apps for companies. <a href="mailto: [email protected]">Email me</a> for business inquiries.
""",
                     inputs=gr.Image(type="pil"), 
                     outputs="text")
iface.launch()