Update app.py

app.py

@@ -11,7 +11,7 @@ def query(payload):
     headers = {"Authorization": f"Bearer {api_token}"}
     API_URL = f"https://api-inference.huggingface.co/models/{model_id}"
     response = requests.post(API_URL, headers=headers, json=payload)
-    return response.json()
+    return response.json(), response
 
 
 
@@ -34,7 +34,7 @@ with placeholder.container():
 
 message(input, is_user=True)  # align's the message to the right
 
-data = query(
+data, resp = query(
     {
         "inputs": {
             "question": input,
@@ -42,11 +42,11 @@ data = query(
         }
     }
 )
-#try:
-model_answer = data["answer"]
-response_templates = [f"{model_answer} is the best task for this 🤩", f"I think you should use {model_answer} 🪄", f"I think {model_answer} should work for you 🤓"]
-bot_answer = random.choice(response_templates)
-message_history.append(bot_answer)
+if resp.status_code == 200:
+    model_answer = data["answer"]
+    response_templates = [f"{model_answer} is the best task for this 🤩", f"I think you should use {model_answer} 🪄", f"I think {model_answer} should work for you 🤓"]
+    bot_answer = random.choice(response_templates)
+    message_history.append(bot_answer)
     
     
 """