Update app.py

app.py

@@ -126,12 +126,12 @@ with st.form(key='my_form'):
         best_probabilities = probabilities[best_indices].tolist()
         words = []              
         st.write(best_words)
-      if submit_button2:
-          print("----")
-          st.write("___")
-          m = BestProbs(prompt)
-          st.write(m)
-      if submit_button3:
-          print("----")
-          st.write("___")
-          st.write(BestProbs)
+    if submit_button2:
+        print("----")
+        st.write("___")
+        m = BestProbs(prompt)
+        st.write(m)
+    if submit_button3:
+        print("----")
+        st.write("___")
+        st.write(BestProbs)