error handling in inference

infer.py

@@ -6,13 +6,17 @@ API_URL = (
 )
 
 
-
-
-
 def query(question, hub_token: str):
     payload = {
         "inputs": question,
+        "parameters": {
+            "wait_for_model": True,
+            "return_full_text": False,
+        },
     }
     headers = {"Authorization": f"Bearer {hub_token}"}
     response = requests.post(API_URL, headers=headers, json=payload)
-    return response.json()[0]["generated_text"]
+    try:
+        return response.json()[0]["generated_text"]
+    except Exception:
+        return "Error occurred while querying the model."