beam

app.py

@@ -19,6 +19,14 @@ text = st.text_area('Enter text and press ctrl/command+enter:')
 if text:
 
     words = [normalize(x) for x in text.split()]
-    outputs = pipe(words, max_length=200)
-    output_text = ' '.join([x['generated_text'] for x in outputs])
-    st.write(output_text)
+    outputs = pipe(words, max_length=200, num_beams=5, num_return_sequences=5)
+
+    texts = [
+        ' '.join(x['generated_text'] for x in option)
+        for option in zip(*outputs)
+    ]
+
+    st.write(texts[0])
+    st.write('Other options:')
+    for option in texts[1:]:
+        st.write(option)