Update app.py

app.py

@@ -16,10 +16,14 @@ model = TFGPT2LMHeadModel.from_pretrained("gpt2", pad_token_id=tokenizer.eos_tok
 def func(sentence, max_length, temperature):
     input_ids = tokenizer.encode(sentence, return_tensors='tf')
     output_list = model.generate(
-        input_ids,
+        input_ids, 
+        do_sample=True, 
+        max_length=max_length, 
+        top_p=0.92, 
+        top_k=0
     )
     output_strs = [tokenizer.decode(output, skip_special_tokens=True) for output in output_list]
-    return output_strs
+    return output_strs[0]
     
 
 demo = gr.Interface(fn=func, inputs=["text", gr.Slider(5, 25), gr.Slider(0.1, 100)], outputs="text")