SaviAnna commited on
Commit
aabd4e2
·
verified ·
1 Parent(s): 63d047a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
  # Функция для генерации текста