Update app.py

app.py

@@ -86,8 +86,10 @@ def BestProbs(prompt):
   for i in best_words[0:10]:
     print("_______")
     st.write(f"${i} $\n")
+    f = (f"${i} $\n")
     m = (prompt + f"{i}")
     BestProbs2(m)
+  return f
 
 def BestProbs2(prompt):
   prompt = prompt.strip()
@@ -123,4 +125,6 @@ with st.form(key='my_form'):
         words = []              
         st.write(best_words)
       if submit_button2:
-          BestProbs(prompt)
+          st.write("___")
+          ok = BestProbs(prompt)
+          st.write(ok)