Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 #from transformers import 
 import tensorflow as tf
 from transformers import TFGPT2LMHeadModel, GPT2Tokenizer
+import math
 
 #generator = pipeline('text-generation', model='gpt2')
 
@@ -18,15 +19,17 @@ def func(sentence, max_length, temperature):
     output_list = model.generate(
         input_ids, 
         do_sample=True, 
-        max_length=max_length, 
+        max_length=Math.floor(max_length),
+        temperature=temperature,
         top_p=0.92, 
         top_k=0
+        num_return_sequences=5
     )
     output_strs = [tokenizer.decode(output, skip_special_tokens=True) for output in output_list]
-    return output_strs[0]
+    return output_strs
     
 
-demo = gr.Interface(fn=func, inputs=["text", gr.Slider(5, 25), gr.Slider(0.1, 100)], outputs="text")
+demo = gr.Interface(fn=func, inputs=["text", gr.Slider(5, 25), gr.Slider(0.1, 100)], outputs=["text", "text", "text", "text", "text"])
 
 if __name__ == "__main__":
     demo.launch()