Added button and error messages

app.py

@@ -12,7 +12,13 @@ def translate(text):
         "inputs": text,
     }
     response = requests.post(API_URL, headers=headers, json=payload)
-    return response.json()
+    json_response = response.json()
+    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."
 
 
 st.set_page_config(
@@ -29,11 +35,11 @@ text = st.text_area("Enter text:",
     help="Enter your text here",
 )
 
-if str(text).strip() == "":
-    st.warning('Please **enter text** for translation')
-else:
-    st.info(str(translate(text)))
-    st.balloons()
+if st.button('Translate'):
+    if str(text).strip() == "":
+        st.warning('Please **enter text** for translation')
+    else:
+        st.info(str(translate(text)))