sees

app.py

@@ -1,16 +1,15 @@
 import os
 import gradio as gr
-from transformers import DistilGPT2Tokenizer, DistilGPT2LMHeadModel
+from transformers import GPT2Tokenizer, GPT2LMHeadModel
 import torch
 
 # Импортируем токены из переменных окружения
 HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
-TG_TOKEN = os.getenv("TG_TOKEN")
 
 # Загрузка модели и токенизатора с использованием токена
-model_name = "distilgpt2"
-tokenizer = DistilGPT2Tokenizer.from_pretrained(model_name, use_auth_token=HUGGINGFACE_TOKEN)
-model = DistilGPT2LMHeadModel.from_pretrained(model_name, use_auth_token=HUGGINGFACE_TOKEN)
+model_name = "gpt2"  # Или другой, если нужно
+tokenizer = GPT2Tokenizer.from_pretrained(model_name, use_auth_token=HUGGINGFACE_TOKEN)
+model = GPT2LMHeadModel.from_pretrained(model_name, use_auth_token=HUGGINGFACE_TOKEN)
 
 # Функция для ведения диалога
 def chat_with_model(user_input):
@@ -24,7 +23,7 @@ iface = gr.Interface(
     fn=chat_with_model,
     inputs="text",
     outputs="text",
-    title="Чатбот на DistilGPT",
+    title="Чатбот на GPT-2",
     description="Поболтай со своим чатботом!"
 )