Update app.py

app.py

@@ -15,32 +15,23 @@ ZEPHYR_API_URL = "https://api-inference.huggingface.co/models/HuggingFaceH4/zeph
 SD_API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
 
 def query_zephyr(linkedin_text):
-    # Construct the prompt to include the LinkedIn post with an instruction for Zephyr
     instruction = "Prepare a prompt for Stable Diffusion for the following LinkedIn post:"
-    conversation = [
-        {"role": "system", "content": instruction},
-        {"role": "user", "content": linkedin_text}
-    ]
+    payload = {
+        "inputs": {
+            "text": instruction + " " + linkedin_text  # Assuming direct instruction followed by the query
+        }
+    }
     
     headers = {
         "Authorization": f"Bearer {ZEPHYR_API_TOKEN}",
         "Content-Type": "application/json",
     }
     
-    # Format the payload according to the Hugging Face Inference API documentation
-    payload = {
-        "inputs": {
-            "past_user_inputs": [],
-            "generated_responses": [],
-            "text": linkedin_text,
-            "conversation": conversation,
-        },
-    }
-    
     response = requests.post(ZEPHYR_API_URL, headers=headers, json=payload)
     if response.status_code == 200:
         return response.json()
     else:
+        print(response.text)  # To get more insight into what went wrong
         raise Exception(f"Failed to query Zephyr model, status code: {response.status_code}")