Update app.py

app.py

@@ -16,7 +16,7 @@ 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: {linkedin_text}"
+    prompt = f"Prepare a prompt for Stable Diffusion for the following LinkedIn post:"
 
     headers = {
         "Authorization": f"Bearer {ZEPHYR_API_TOKEN}",
@@ -25,7 +25,9 @@ def query_zephyr(linkedin_text):
 
     # The `inputs` field now directly contains the prompt string
     payload = {
-        "inputs": prompt,
+
+        "role": "system", "content": prompt,
+        "role": "user", "content": linkedin_text
     }
 
     response = requests.post(ZEPHYR_API_URL, headers=headers, json=payload)