Update app.py

app.py

@@ -47,7 +47,7 @@ def change_model_name(name):
     MODEL_BUF["config"] = AutoConfig.from_pretrained(name)
 
 
-def generate(text, model, tokenizer):
+def generate(model, text):
    model.eval()
    input_ids = tokenizer.encode("AFA:{}".format(text), return_tensors="pt")
    outputs = model.generate(input_ids, max_length=200, num_beams=2, repetition_penalty=2.5, top_k=50, top_p=0.98, length_penalty=1.0, early_stopping=True)