added error msg

app.py

@@ -84,7 +84,11 @@ def main():
         else:
             # render masked language modeling table
             mlm_result = mask_api(inputs=user_input)
-            # st.write(mlm_result)
+
+            if mlm_result == None:
+                st.write("Model is loading. Please try again later...")
+                return
+
             mlm_df = pd.DataFrame(mlm_result)
             mlm_df.drop(columns=["token", "token_str"], inplace=True)
             mlm_df_styled = mlm_df.copy(deep=False)
@@ -109,4 +113,5 @@ def main():
             display_table(sa_df_styled, "🤔 By saying that, I guess you are feeling..")
 
 
-main()
+if __name__ == "__main__":
+    main()