Update README.md
Browse files
README.md
CHANGED
|
@@ -42,4 +42,11 @@ SberGPT2 InternetLanguage-RU Model (Experimental)
|
|
| 42 |
```python
|
| 43 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 44 |
|
| 45 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |
```python
|
| 43 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 44 |
|
| 45 |
+
tokenizer = AutoTokenizer.from_pretrained("MarkProMaster229/InternetLanguage")
|
| 46 |
+
model = AutoModelForCausalLM.from_pretrained("MarkProMaster229/InternetLanguage")
|
| 47 |
+
|
| 48 |
+
# Пример генерации текста
|
| 49 |
+
input_text = "Привет! Как дела?"
|
| 50 |
+
inputs = tokenizer(input_text, return_tensors="pt")
|
| 51 |
+
outputs = model.generate(**inputs, max_length=50, do_sample=True, top_k=50, top_p=0.95)
|
| 52 |
+
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|