beam

app.py

@@ -13,6 +13,8 @@ def generate_text(prompt, max_length, temperature):
         generated = generator(
             prompt,
             max_length=max_length,
+            #num_beams=10,
+            #no_repeat_ngram_size=2,
             temperature=temperature,
             do_sample=True,
             repetition_penalty=1.4