Update app.py

app.py

@@ -1,11 +1,11 @@
 
-from transformers import AutoTokenizer, AutoModelWithLMHead
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import torch
 import gradio as gr
 
-# Загружаем токенизатор и модель
+# Загружаем модель rut5-base с медленным токенизатором
 model_name = "cointegrated/rut5-base"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
 # Функция генерации ответа
@@ -29,7 +29,7 @@ def generate_response(prompt):
 iface = gr.Interface(
     fn=generate_response,
     inputs=gr.Textbox(lines=4, label="Введите ваш запрос"),
-    outputs=gr.Textbox(label="Ответ от ChatGPT"),
+    outputs=gr.Textbox(label="Ответ модели"),
     title="Интерфейс ChatGPT",
     description="Пример взаимодействия с API OpenAI через Hugging Face Space"
 )