Update app.py

app.py

@@ -9,7 +9,7 @@ model = AutoDistributedModelForCausalLM.from_pretrained(model_name)
 
 import gradio as gr
 
-def generate(input, params):
+def generate(input):
     tokenized = tokenizer(input, return_tensors="pt")["input_ids"]
     outputs = model.generate(inputs, max_new_tokens=80, do_sample=True, temperature=0.9)
     return tokenizer.decode(outputs[0]).replace("</s>", "");