Update README.md

README.md

@@ -40,7 +40,7 @@ def load_model(model_name):
 def generate_text(prompt, model, tokenizer):
     inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
     with torch.no_grad():
-        generated_code = tokenizer.decode(model.generate(**inputs, max_new_tokens=1024, pad_token_id=2)[0], skip_special_tokens=True)
+        generated_code = tokenizer.decode(model.generate(**inputs, max_new_tokens=256, pad_token_id=2)[0], skip_special_tokens=True)
     print(generated_code)
     return generated_code
 
@@ -50,7 +50,6 @@ model, tokenizer = load_model(model_name)
 
 prompt = "একটি গ্রামের বর্ণনা করুন।"
 generated_text = generate_text(prompt, model, tokenizer)
-print(generated_text)
 ```
 
 ## Training Details