Update app.py

app.py

@@ -52,9 +52,7 @@ def generate_text(
         logging.info(f"Input too long {input_len} > {abs_max_length}, may cause errors")
     result = generator(
         prompt,
-        # max_length=gen_length + input_len, # old API for generation
-        max_length=None, 
-        max_new_tokens=gen_length, 
+        max_length=gen_length + input_len, # old API for generation
         min_length=input_len + 4,
         penalty_alpha=penalty_alpha,
         top_k=top_k,