Update README.md

README.md

@@ -45,14 +45,14 @@ input_prompts = [
 # Generate responses
 generated_responses = {}
 for prompt in input_prompts:
-    inputs = tokenizer(prompt, return_tensors="pt", max_length=400, truncation=True, padding="max_length").to(device)
+    inputs = tokenizer(prompt, return_tensors="pt", max_length=40, truncation=True, padding="max_length").to(device)
     
     model.eval()
     with torch.no_grad():
         generated_ids = model.generate(
             input_ids=inputs['input_ids'],
             attention_mask=inputs['attention_mask'],
-            max_length=40,
+            max_length=100,
             num_beams=7,
             repetition_penalty=2.5,
             length_penalty=2.0,
@@ -66,4 +66,4 @@ for prompt in input_prompts:
 # Display the input prompts and the generated responses
 for prompt, response in generated_responses.items():
     print(f"Prompt: {prompt}")
-    print(f"Response: {response}\n")
+    print(f"Response: {response}\n")