Update app.py

app.py

@@ -7,7 +7,7 @@ pipe = transformers.pipeline("text-generation",
 pipe.to('cuda')
 
 @spaces.GPU
-def infer(prompt, max_tokens):
+def infer(prompt):
     return pipe(prompt, max_new_tokens=1, do_sample=True, temperature=1.0)[0]["generated_text"]
 
 demo = gr.Interface(