Update app.py

app.py

@@ -99,9 +99,9 @@ def BestProbs2(prompt):
   logits, past_key_values = model(myinput, past_key_values = past_key_values, return_dict=False)
   logits = logits[0,-1]
   probabilities = torch.nn.functional.softmax(logits)
-  best_logits, best_indices = logits.topk(10)
+  best_logits, best_indices = logits.topk(20)
   best_words = [tokenizer.decode([idx.item()]) for idx in best_indices]
-  for i in best_words[0:10]:
+  for i in best_words[0:20]:
     print(i)
     st.write(i)
 
@@ -130,7 +130,9 @@ with st.form(key='my_form'):
         print("----")
         st.write("___")
         m = BestProbs(prompt)
+        st.write("___")
         st.write(m)
+        st.write("___")
     if submit_button3:
         print("----")
         st.write("___")