Update app.py

app.py

@@ -3,6 +3,34 @@ 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 = """