Update app.py

app.py

@@ -14,30 +14,36 @@ MODEL_ENDPOINTS = {
 
 def query_model(prompt, model_endpoint):
     """
-    Query a model via Hugging Face Inference API using a requests.post call.
-    This assumes an OpenAI-compatible endpoint structure.
+    Query a model via the Hugging Face Inference API using a requests.post call.
+    This payload now uses the 'inputs' field as required.
     """
     headers = {
         "Authorization": f"Bearer {HF_API_KEY}",
         "Content-Type": "application/json"
     }
+    # Use 'inputs' instead of 'messages' and pass parameters inside the 'parameters' field.
     data = {
-        "messages": [{"role": "user", "content": prompt}],
-        "max_tokens": 512,
-        "temperature": 0.7
+        "inputs": prompt,
+        "parameters": {
+            "max_new_tokens": 512,
+            "temperature": 0.7,
+        }
     }
     
     response = requests.post(model_endpoint, headers=headers, json=data)
+    
     try:
         result = response.json()
     except Exception:
         return f"Error: Unable to parse JSON. Response: {response.text}"
     
-    if "error" in result:
+    # If there's an error message in the result, return it.
+    if isinstance(result, dict) and "error" in result:
         return f"Error: {result['error']}"
     
     try:
-        return result["choices"][0]["message"]["content"]
+        # Expected result format is a list of outputs with a "generated_text" field.
+        return result[0]["generated_text"]
     except Exception:
         return f"Error: Unexpected response format: {result}"