Update app.py

app.py

@@ -20,13 +20,12 @@ def generate_text(text):
         # Generate text
         output = model.generate(
             input_ids=encoded_input['input_ids'],
-            max_length=100,  # Specify the max length for the generated text
+            max_length=200,  # Specify the max length for the generated text
             num_return_sequences=1,  # Number of sequences to generate
             no_repeat_ngram_size=2,  # Avoid repeating n-grams of length 2
             top_k=50,  # Limits the sampling pool to top_k tokens
             top_p=0.95,  # Cumulative probability threshold for nucleus sampling
             temperature=0.7,  # Controls the randomness of predictions
-            do_sample=True,  # Enable sampling
             attention_mask=encoded_input['attention_mask'],  # Correct attention mask
             pad_token_id=tokenizer.eos_token_id  # Use the end-of-sequence token as padding
         )