Adding try except

app.py

@@ -11,14 +11,15 @@ def translate(text):
     payload  = {
         "inputs": text,
     }
-    response = requests.post(API_URL, headers=headers, json=payload)
-    json_response = response.json()[0]
-    if "error" in json_response:
-        return f"Model still loading: {json_response}"
-    elif "translation_text" in json_response:
-        return json_response["translation_text"]
-    else:
-        return "Oops, something went wrong."
+    try:
+        response = requests.post(API_URL, headers=headers, json=payload)
+        json_response = response.json()[0]
+        if "error" in json_response:
+            return f"Model still loading: {json_response}"
+        elif "translation_text" in json_response:
+            return json_response["translation_text"]
+    except Exception as exc:
+        return f"Oops, something went wrong: {exc}"
 
 
 st.set_page_config(