tonyassi's picture
Update app.py
34b0d02
raw
history blame
1.36 kB
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']
#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](https://huggingface.co/tonyassi/fashion-clothing-decade) image classification model. Please ❤️ this Space.
I build custom AI apps for companies. <a href="mailto: [email protected]">Email me</a> for business inquiries.
![](https://cdn.discordapp.com/attachments/1120417968032063538/1184251611388850257/all.png?ex=658b4b42&is=6578d642&hm=6b853c5b1e92d07701496f5fcb3106c6cef15c66a1a238ceedb214b4d9348245&)
""",
inputs=gr.Image(type="pil"),
outputs=gr.Markdown(),
examples=[['./examples/1910s.jpg'],['./examples/1930s.jpg'],['./examples/1940s.jpg'],['./examples/1950s.jpg'],['./examples/1960s.jpg'],['./examples/1980s.jpg'],['./examples/1990s.jpg'],['./examples/2000s.jpg'],]
)
iface.launch()