fn args

app.py

@@ -4,12 +4,11 @@ from transformers import pipeline, set_seed, AutoTokenizer
 
 set_seed(42)
 
-tokenizer = AutoTokenizer.from_pretrained("bert-base-cased")
-generator = pipeline('text-generation', model='gpt2', return_full_text=False, tokenizer=tokenizer)
+generator = pipeline('text-generation', model='gpt2', return_full_text=False)
 
 
-def Bemenet(bemenet):
-    return generator(os.environ['SECRET_PROMPT'] + " " + bemenet, max_length=100, num_return_sequences=2)[1]['generated_text']
+def Bemenet(input_string):
+    return generator(os.environ['SECRET_PROMPT'] + " " + input_string, max_length=100, num_return_sequences=2)[1]['generated_text']
 
 textbox = gr.Textbox()