svilanovich commited on
Commit
d15958c
·
verified ·
1 Parent(s): 70488ed

Update pages/ruGPT.py

Browse files
Files changed (1) hide show
  1. pages/ruGPT.py +1 -1
pages/ruGPT.py CHANGED
@@ -32,7 +32,7 @@ def generate_text(text: str,
32
 
33
 
34
 
35
- paath = 'C:\\Users\\igors\\elbrus_text_project\\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)
 
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)