Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ def load_gpt2():
|
|
23 |
def load_custom_model():
|
24 |
# Здесь замените путь на вашу кастомную модель
|
25 |
model_custom = GPT2LMHeadModel.from_pretrained("./rus_gpt2_tuned", from_tf=False, use_safetensors=True)
|
26 |
-
tokenizer_custom = GPT2TokenizerFast.from_pretrained("./rus_gpt2_tuned")
|
27 |
return model_custom, tokenizer_custom
|
28 |
|
29 |
# Функция для генерации текста
|
|
|
23 |
def load_custom_model():
|
24 |
# Здесь замените путь на вашу кастомную модель
|
25 |
model_custom = GPT2LMHeadModel.from_pretrained("./rus_gpt2_tuned", from_tf=False, use_safetensors=True)
|
26 |
+
tokenizer_custom = GPT2TokenizerFast.from_pretrained("./rus_gpt2_tuned/tokenizer")
|
27 |
return model_custom, tokenizer_custom
|
28 |
|
29 |
# Функция для генерации текста
|