Update app.py

app.py

@@ -120,7 +120,7 @@ def stream_response(prompt_text, api_key):
     Yields each chunk of text as it arrives.
     """
     try:
-        # Match the structure of your working payload:
+    # Match the structure of your working payload:
         payload = {
             "inputs": prompt_text,
             "parameters": {
@@ -129,14 +129,19 @@ def stream_response(prompt_text, api_key):
                 "stream": True
             }
         }
+        headers = {
+                	"Accept" : "application/json",
+                    "Authorization": f"Bearer {api_key}",
+                	"Content-Type": "application/json" 
+                  }
 
         # POST request with stream=True to get partial chunks
         response = requests.post(
             SPACE_URL,
             json=payload,
-            headers={"Authorization": f"Bearer {api_key}"},
+            headers=headers,
             stream=True
-        )
+        ))
         response.raise_for_status()
 
         # The endpoint presumably returns lines of JSON. Adjust parsing if needed: