Update README.md
Browse files
README.md
CHANGED
|
@@ -56,7 +56,7 @@ inputs = tokenizer.encode(prompt_urdu)
|
|
| 56 |
inputs_tensor = torch.tensor(inputs).unsqueeze(0) # Add batch dimension
|
| 57 |
|
| 58 |
# Generate text
|
| 59 |
-
outputs = model.generate(inputs_tensor, max_new_tokens=
|
| 60 |
outputs_tensor = torch.tensor(outputs).unsqueeze(0)
|
| 61 |
generated_text = tokenizer.decode(outputs_tensor[0].squeeze().tolist())
|
| 62 |
|
|
|
|
| 56 |
inputs_tensor = torch.tensor(inputs).unsqueeze(0) # Add batch dimension
|
| 57 |
|
| 58 |
# Generate text
|
| 59 |
+
outputs = model.generate(inputs_tensor, max_new_tokens=64, temperature=0.7)
|
| 60 |
outputs_tensor = torch.tensor(outputs).unsqueeze(0)
|
| 61 |
generated_text = tokenizer.decode(outputs_tensor[0].squeeze().tolist())
|
| 62 |
|