Spaces:
Sleeping
Sleeping
Update pages/ruGPT.py
Browse files- pages/ruGPT.py +1 -1
pages/ruGPT.py
CHANGED
|
@@ -32,7 +32,7 @@ def generate_text(text: str,
|
|
| 32 |
|
| 33 |
|
| 34 |
|
| 35 |
-
paath = '
|
| 36 |
model = GPT2LMHeadModel.from_pretrained(paath).to(device)
|
| 37 |
model_name_or_path = "ai-forever/rugpt3small_based_on_gpt2"
|
| 38 |
tokenizer = GPT2Tokenizer.from_pretrained(model_name_or_path)
|
|
|
|
| 32 |
|
| 33 |
|
| 34 |
|
| 35 |
+
paath = 'data/finetuned/'
|
| 36 |
model = GPT2LMHeadModel.from_pretrained(paath).to(device)
|
| 37 |
model_name_or_path = "ai-forever/rugpt3small_based_on_gpt2"
|
| 38 |
tokenizer = GPT2Tokenizer.from_pretrained(model_name_or_path)
|