Update app.py

app.py

@@ -16,19 +16,26 @@ SD_API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-dif
 
 def query_zephyr(linkedin_text):
     # Constructing a single string prompt that combines the instruction with the LinkedIn post text
-    prompt = f"Prepare a prompt for Stable Diffusion for the following LinkedIn post:"
-
+    
     headers = {
         "Authorization": f"Bearer {ZEPHYR_API_TOKEN}",
         "Content-Type": "application/json",
     }
 
-    # The `inputs` field now directly contains the prompt string
-    payload = {
-
-        "role": "system", "content": prompt,
-        "role": "user", "content": linkedin_text
+   prompt = "Prepare a prompt for Stable Diffusion for the following LinkedIn post:"
+conversation = [
+    {"role": "system", "content": prompt},
+    {"role": "user", "content": linkedin_text}
+]
+
+payload = {
+    "inputs": {
+        "past_user_inputs": [],
+        "generated_responses": [],
+        "text": linkedin_text,  # This might need adjustment based on the model's requirements.
+        "conversation": conversation,
     }
+}
 
     response = requests.post(ZEPHYR_API_URL, headers=headers, json=payload)
     if response.status_code == 200: