Update app.py

app.py

@@ -87,6 +87,7 @@ Translated into the Style of Abraham Lincoln: corn fields ( permeate illinois /
 
 informal english: """
 
+number_of_outputs = st.sidebar.slider("Number of Outputs", 5, 20)
 
 def BestProbs(prompt):
   prompt = prompt.strip()
@@ -175,9 +176,9 @@ def BestProbs5(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(5)
+  best_logits, best_indices = logits.topk(number_of_outputs)
   best_words = [tokenizer.decode([idx.item()]) for idx in best_indices]
-  for i in best_words[0:5]:
+  for i in best_words[0:number_of_outputs]:
     #print(i)
     print("\n")
     g = (prompt + i)