Spaces:
Running
Running
| import gradio as gr | |
| from transformers import pipeline | |
| decade = pipeline(model="tonyassi/fashion-clothing-decade") | |
| def mistral(caption): | |
| url = os.environ.get('MISTRAL_URL') | |
| # Define the prompt | |
| prompt = "given the following caption please provide some fashion styling advice: " + caption | |
| # Mistral API call | |
| payload = json.dumps({ | |
| "key": os.environ.get('MISTRAL_KEY'), | |
| "messages": [ | |
| { | |
| "role": "user", | |
| "content": prompt | |
| }, | |
| ], | |
| "max_tokens": 1000 | |
| }) | |
| headers = { | |
| 'Content-Type': 'application/json' | |
| } | |
| # API response | |
| response = requests.request("POST", url, headers=headers, data=payload) | |
| response = json.loads(response.text) | |
| return response['message'] | |
| 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. | |
|  | |
| """, | |
| inputs=gr.Image(type="pil"), | |
| outputs=gr.Markdown(), | |
| examples=[['./examples/1910s.jpg'],['./examples/1920s.jpg'],['./examples/1930s.jpg'],['./examples/1940s.jpg'],['./examples/1950s.jpg'],['./examples/1960s.jpg'],['./examples/1970s.jpg'],['./examples/1980s.jpg'],['./examples/1990s.jpg'],['./examples/2000s.jpg'],] | |
| ) | |
| iface.launch() |