eeea

app.py

@@ -9,8 +9,8 @@ HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
 
 # Загрузка модели и токенизатора DialoGPT
 model_name = "microsoft/DialoGPT-medium"  # Можно использовать small или large версии
-tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=HUGGINGFACE_TOKEN, padding_side='left')
-model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=HUGGINGFACE_TOKEN)
+tokenizer = AutoTokenizer.from_pretrained(model_name, token=HUGGINGFACE_TOKEN, padding_side='left')
+model = AutoModelForCausalLM.from_pretrained(model_name, token=HUGGINGFACE_TOKEN)
 
 # Список саркастических фраз
 sarcastic_phrases = [
@@ -51,4 +51,20 @@ def chat_with_model(user_input, chat_history=[]):
     if random.random() < 0.5:  # 50% шанс на сарказм
         bot_response += " " + random.choice(sarcastic_phrases)
 
-    # Обновляем истори
+    # Обновляем историю
+    chat_history.append(new_user_input_ids)
+    chat_history.append(chat_history_ids[:, bot_input_ids.shape[-1]:])
+
+    return bot_response, chat_history
+
+# Создание интерфейса Gradio
+iface = gr.Interface(
+    fn=chat_with_model,
+    inputs="text",
+    outputs="text",
+    title="Чатбот на DialoGPT с сарказмом",
+    description="Поболтай со своим саркастичным чатботом!"
+)
+
+# Запуск интерфейса
+iface.launch()