MarkProMaster229 commited on
Commit
6820768
·
verified ·
1 Parent(s): e540ee3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -1
README.md CHANGED
@@ -42,4 +42,11 @@ SberGPT2 InternetLanguage-RU Model (Experimental)
42
  ```python
43
  from transformers import AutoTokenizer, AutoModelForCausalLM
44
 
45
- model = AutoModelForCausalLM.from_pretrained("MarkProMaster229/InternetLanguage"
 
 
 
 
 
 
 
 
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))