add error messsage if still loading

app.py

@@ -73,7 +73,7 @@ top_k = st.sidebar.text_input(
 
 top_p = st.sidebar.text_input(
     "Top p",
-    value=1.0,
+    value=0.95,
     help=" If set to float < 1, only the most probable tokens with probabilities that add up to top_p or higher are kept for generation."
 )
 
@@ -105,8 +105,12 @@ if st.button("Run"):
                          temp=temp,
                          top_k=int(top_k),
                          top_p=float(top_p))
+
         print("result:", result)
-        result = result[0]["generated_text"]
-        st.write(result.replace("\n", "  \n"))
-        st.text("English translation")
-        st.write(translate(result, "en", "id").replace("\n", "  \n"))
+        if "error" in result:
+            st.write(f'{result["error"]}. Please try it again in about {result["estimated_time"]:.0f} seconds')
+        else:
+            result = result[0]["generated_text"]
+            st.write(result.replace("\n", "  \n"))
+            st.text("English translation")
+            st.write(translate(result, "en", "id").replace("\n", "  \n"))