Update app.py

app.py

@@ -22,8 +22,8 @@ def load_gpt2():
 @st.cache_resource
 def load_custom_model():
     # Здесь замените путь на вашу кастомную модель
-    model_custom = GPT2LMHeadModel.from_pretrained("/rus_gpt2_tuned")
-    tokenizer_custom = GPT2TokenizerFast.from_pretrained("/rus_gpt2_tuned")
+    model_custom = GPT2LMHeadModel.from_pretrained("./rus_gpt2_tuned", from_tf=False, use_safetensors=True))
+    tokenizer_custom = GPT2TokenizerFast.from_pretrained("./rus_gpt2_tuned")
     return model_custom, tokenizer_custom
 
 # Функция для генерации текста