few modifs

app.py

@@ -23,8 +23,8 @@ def generate_single(model, tokenizer, input_sentence, max_length=50, top_k=0, te
     generated_sentence = tokenizer.decode(output[0], skip_special_tokens=True)
     return post_process_sentence(input_sentence, generated_sentence)
 
-def question_bloom(input_sentence, max_length, temperature, do_sample=True, seed=42):
-    post_processed_output = generate_single(model, tokenizer, input_sentence, temperature=temperature, max_length=max_length, do_sample=do_sample, seed=seed)
+def question_bloom(input_sentence, max_length, temperature, do_sample=True, top_k=3, seed=42):
+    post_processed_output = generate_single(model, tokenizer, input_sentence, temperature=temperature, max_length=max_length, do_sample=do_sample, top_k=top_k, seed=seed)
     return post_processed_output.split("\n-")[-2]
 
 gr.Interface(
@@ -46,6 +46,13 @@ gr.Interface(
             label="Temperature",
         ),
         gr.inputs.Checkbox(True, label="Do Sample"),
+        gr.inputs.Slider(
+            minimum=0,
+            maximum=10,
+            step=1,
+            default=3,
+            label="Top K",
+        ),
         gr.inputs.Slider(
             minimum=0,
             maximum=256,