Update app.py

app.py

@@ -1,13 +1,15 @@
 import gradio as gr
 from transformers import pipeline
+import requests
+import json
 
 decade = pipeline(model="tonyassi/fashion-clothing-decade")
 
-def mistral(caption):
+def mistral(decade):
     url = os.environ.get('MISTRAL_URL')
     
     # Define the prompt
-    prompt = "given the following caption please provide some fashion styling advice: " + caption
+    prompt = "write a fun,short description of the " + decade + " in 1-2 sentences"
 
     # Mistral API call
     payload = json.dumps({
@@ -34,7 +36,8 @@ def mistral(caption):
 def greet(img):
     pred = decade(images=img)
     res = """
-# """ + pred[0]['label']
+# """ + pred[0]['label'] + """
+""" + mistral(pred[0]['label'])
     #res += pred[1]['label'] + ':' + str(pred[1]['score']) + '\n'
     #res += pred[2]['label'] + ':' + str(pred[2]['score'])
     return res