Update README.md
Browse files
README.md
CHANGED
|
@@ -63,7 +63,7 @@ inputs_tensor = torch.tensor(inputs).unsqueeze(0) # Add batch dimension
|
|
| 63 |
|
| 64 |
# Generate text
|
| 65 |
outputs = model.generate(inputs_tensor, max_new_tokens=128, temperature=0.7)
|
| 66 |
-
outputs_tensor = torch.tensor(
|
| 67 |
generated_text = tokenizer.decode(outputs_tensor[0].squeeze().tolist())
|
| 68 |
|
| 69 |
print(f"Prompt: {prompt_urdu}")
|
|
|
|
| 63 |
|
| 64 |
# Generate text
|
| 65 |
outputs = model.generate(inputs_tensor, max_new_tokens=128, temperature=0.7)
|
| 66 |
+
outputs_tensor = torch.tensor(outputs).unsqueeze(0)
|
| 67 |
generated_text = tokenizer.decode(outputs_tensor[0].squeeze().tolist())
|
| 68 |
|
| 69 |
print(f"Prompt: {prompt_urdu}")
|