Update model_api.py

model_api.py

@@ -14,21 +14,24 @@ def query_model(prompt):
     payload = {
         "inputs": prompt,
         "parameters": {
-            "max_new_tokens": 800,
+            "max_new_tokens": 500,
             "temperature": 0.7
         }
     }
 
-    while True:
-        response = requests.post(API_URL, headers=headers, json=payload)
+    response = requests.post(API_URL, headers=headers, json=payload)
+
+    # ✅ Check status code first
+    if response.status_code != 200:
+        return f"HTTP Error {response.status_code}: {response.text}"
+
+    # ✅ Safe JSON parsing
+    try:
         result = response.json()
+    except Exception:
+        return f"Invalid response received: {response.text}"
 
-        # Success response
-        if isinstance(result, list):
-            return result[0]["generated_text"]
+    if isinstance(result, list):
+        return result[0]["generated_text"]
 
-        # Model loading
-        if "estimated_time" in result:
-            time.sleep(result["estimated_time"])
-        else:
-            return f"Error: {result}"
+    return f"Unexpected response: {result}"