Update app.py

app.py

@@ -37,18 +37,21 @@ RETRY_COMMAND = "/retry"
 STOP_SEQ = [f"\n{USER_NAME}", "<|end|>"]
 
 def run_model(prompt, model, temperature, top_p):
-    api_url = f"https://api-inference.huggingface.co/models/{model}"
-    payload = {
-        "inputs": prompt,
-        "parameters": {
-            "max_new_tokens": 128,
-            "do_sample": True,
-            "temperature": temperature,
-            "top_p": top_p
+    try:
+        api_url = f"https://api-inference.huggingface.co/models/{model}"
+        payload = {
+            "inputs": prompt,
+            "parameters": {
+                "max_new_tokens": 128,
+                "do_sample": True,
+                "temperature": temperature,
+                "top_p": top_p
+            }
         }
-    }
-    response = requests.post(api_url, headers=HEADERS, json=payload)
-    return response.json()[0]['generated_text']
+        response = requests.post(api_url, headers=HEADERS, json=payload)
+        return response.json()[0]['generated_text']
+    except:
+        return "I'm sorry, the model is not available right now. Please try again later."
 
 def get_stream(string: str):
     return enumerate(iter(string.split(" ")))