Update app.py

app.py

@@ -42,15 +42,14 @@ def generate_text(model, tokenizer, prompt, max_len, temperature):
     #                        pad_token_id=tokenizer.eos_token_id)
     output = model.generate(
     input_ids,
-    max_length=max_length,
+    max_length=max_len,
     temperature=temperature,              # Controls the diversity of the generated text
     top_k=50,                     # Keeps only the top-k most likely words
     top_p=0.9,                    # Nucleus sampling (cumulative probability)
     repetition_penalty=1.2,       # Penalty for repeating words or phrases
     no_repeat_ngram_size=4,       # Prevents repetition of n-grams (e.g., bigrams)
     do_sample=True,                # Enables sampling for greater diversity
-    pad_token_id=tokenizer.eos_token_id,
-    max_length=max_len)[0]
+    pad_token_id=tokenizer.eos_token_id)[0]
 
 
     #generated_text = tokenizer.decode(output[0], skip_special_tokens=True)