Update README.md
Browse files
README.md
CHANGED
@@ -37,6 +37,7 @@ tags:
|
|
37 |
## Example Usage (Python)
|
38 |
|
39 |
```python
|
|
|
40 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
41 |
|
42 |
model = AutoModelForCausalLM.from_pretrained("lambdaindie/lambdai")
|
@@ -46,7 +47,9 @@ input_text = "Problema: Prove que 17 é um número primo."
|
|
46 |
inputs = tokenizer(input_text, return_tensors="pt")
|
47 |
|
48 |
output = model.generate(**inputs, max_new_tokens=100)
|
49 |
-
print(tokenizer.decode(output[0], skip_special_tokens=True))
|
|
|
|
|
50 |
|
51 |
About Lambda
|
52 |
|
|
|
37 |
## Example Usage (Python)
|
38 |
|
39 |
```python
|
40 |
+
|
41 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
42 |
|
43 |
model = AutoModelForCausalLM.from_pretrained("lambdaindie/lambdai")
|
|
|
47 |
inputs = tokenizer(input_text, return_tensors="pt")
|
48 |
|
49 |
output = model.generate(**inputs, max_new_tokens=100)
|
50 |
+
print(tokenizer.decode(output[0], skip_special_tokens=True))
|
51 |
+
|
52 |
+
```
|
53 |
|
54 |
About Lambda
|
55 |
|