Update application

app.py

@@ -30,9 +30,9 @@ MODELS = {
 PROMPT_LIST = [
     "Fui a la librería a comprar un <mask>.",
     "¡Qué buen <mask> hace hoy!",
-    "Hoy empiezan las vacaciones, vamos a la <mask>.",
+    "Hoy empiezan las vacaciones así que vamos a la <mask>.",
     "Mi color favorito es el <mask>.",
-    "Voy a <mask>, estoy muy cansada.",
+    "Voy a <mask> porque estoy muy cansada.",
     "Mañana vienen mis amigos de <mask>.",
     "¿Te apetece venir a <mask> conmigo?",
     "En verano hace mucho <mask>.",
@@ -77,25 +77,13 @@ else:
 text = st.text_area("Enter text", prompt_box)
 
 if st.button("Fill the mask"):
-    with st.spinner(text="Getting results..."):
+    with st.spinner(text="Filling the mask..."):
         st.subheader("Result")
         result = load_model(text, model_url)
-        if "error" in result:
-            if type(result["error"]) is str:
-                st.write(f'{result["error"]}.', end=" ")
-                if "estimated_time" in result:
-                    st.write(
-                        f'Please try again in about {result["estimated_time"]:.0f} seconds.'
-                    )
-            else:
-                if type(result["error"]) is list:
-                    for error in result["error"]:
-                        st.write(f"{error}")
-        else:
-            result_sequence, result_token = result[0]["sequence"], result[0]["token_str"]
-            st.write(result_sequence)
-            st.text("English translation")
-            st.write(translate(result_sequence, "en", "es"))
+        result_sequence = result[0]["sequence"]
+        st.write(result_sequence)
+        st.write("_English_ _translation:_", translate(result_sequence, "en", "es"))
+        st.write(result)
 
 st.markdown(
     """