Update app.py

app.py

@@ -10,35 +10,28 @@ import spaces
 import warnings
 import time
 
+# Подавляем технические предупреждения
 warnings.filterwarnings("ignore", message="The PyTorch API of nested tensors is in prototype stage")
 warnings.filterwarnings("ignore", category=UserWarning, module="torch.nn.modules.transformer")
 
 # ============ НАСТРОЙКИ ============
 DATA_DIR = '/data'
 os.makedirs(DATA_DIR, exist_ok=True)
-MODEL_BIN = os.path.join(DATA_DIR, 'andrey_real_nn.bin')
-MODEL_GGUF = os.path.join(DATA_DIR, 'andrey_real_nn.gguf')
+MODEL_BIN = os.path.join(DATA_DIR, 'andrey_final.bin')
+MODEL_GGUF = os.path.join(DATA_DIR, 'andrey_final.gguf')
 
 # ============ СЛОВАРЬ ============
 WORDS = [
-    'привет', 'здравствуй', 'добрый', 'день', 'утро', 'вечер', 'ночь', 'рад', 'видеть', 'слышать',
-    'как', 'ты', 'дела', 'жизнь', 'настроение', 'что', 'нового', 'кто', 'где', 'когда', 'почему', 'зачем',
-    'хорошо', 'отлично', 'нормально', 'плохо', 'грустно', 'весело', 'скучно', 'интересно',
-    'спасибо', 'пожалуйста', 'извини', 'конечно', 'да', 'нет', 'возможно', 'наверное',
+    'привет', 'здравствуй', 'добрый', 'день', 'утро', 'вечер', 'ночь', 'рад', 'видеть',
+    'как', 'ты', 'дела', 'жизнь', 'настроение', 'что', 'нового', 'кто', 'где', 'когда', 'почему',
+    'хорошо', 'отлично', 'нормально', 'плохо', 'спасибо', 'пожалуйста', 'да', 'нет',
     'я', 'мы', 'он', 'она', 'думаю', 'знаю', 'хочу', 'могу', 'буду', 'люблю', 'работаю', 'учусь',
-    'читаю', 'пишу', 'говорю', 'слушаю', 'смотрю', 'играю', 'сплю', 'ем', 'иду', 'бегу',
-    'создан', 'живу', 'нахожусь', 'помогаю', 'общаюсь', 'стараюсь', 'понимаю', 'чувствую',
     'андрей', 'бот', 'помощник', 'нейросеть', 'евгений', 'создатель', 'openrussianai', 'тверь', 'россия',
-    'hugging', 'face', 'платформа', 'дом', 'мир', 'люди', 'друг', 'время', 'сегодня', 'завтра', 'вчера',
-    'книга', 'фильм', 'музыка', 'игра', 'работа', 'учеба', 'отдых', 'путешествие', 'еда', 'вода',
-    'солнце', 'луна', 'звезды', 'небо', 'земля', 'море', 'река', 'город', 'деревня', 'лес', 'поле',
+    'hugging', 'face', 'платформа', 'дом', 'мир', 'люди', 'друг', 'время', 'сегодня', 'завтра',
     'плюс', 'минус', 'умножить', 'разделить', 'равно', 'ноль', 'один', 'два', 'три', 'четыре', 'пять',
-    'шесть', 'семь', 'восемь', 'девять', 'десять', 'сто', 'тысяча',
     'и', 'а', 'но', 'или', 'в', 'на', 'с', 'из', 'к', 'у', 'не', 'ли', 'же', 'бы', 'очень', 'уже', 'ещё',
-    'мой', 'твой', 'наш', 'ваш', 'его', 'её', 'их', 'этот', 'тот', 'какой', 'который', 'весь', 'сам',
-    'радость', 'счастье', 'удача', 'успех', 'мечта', 'цель', 'любовь', 'дружба', 'семья', 'здоровье',
-    'добрый', 'злой', 'веселый', 'грустный', 'умный', 'глупый', 'красивый', 'сильный', 'быстрый',
-    'расскажи', 'покажи', 'объясни', 'помоги', 'скажи', 'спой', 'станцуй', 'нарисуй', 'придумай',
+    'мой', 'твой', 'наш', 'этот', 'тот', 'какой', 'который', 'весь', 'сам',
+    'расскажи', 'покажи', 'объясни', 'помоги', 'скажи', 'шутка', 'смех', 'радость', 'счастье', 'удача'
 ]
 
 word_to_idx = {w: i+3 for i, w in enumerate(WORDS)}
@@ -54,13 +47,22 @@ MAX_LEN = 25
 def tokenize(text):
     return [word_to_idx.get(w, UNK) for w in text.lower().split()]
 
+# Жесткая фильтрация спецтокенов при выводе
 def detokenize(tokens):
-    return ' '.join([idx_to_word.get(t, '?') for t in tokens if t not in [PAD, UNK, START]])
+    words = []
+    for t in tokens:
+        t = int(t)
+        if t in [PAD, UNK, START]:
+            continue
+        w = idx_to_word.get(t, '')
+        if w and not w.startswith('['):
+            words.append(w)
+    return ' '.join(words)
 
 def pad_sequence(seq, max_len=MAX_LEN):
     return (seq + [PAD] * max_len)[:max_len]
 
-# ============ АРХИТЕКТУРА TRANSFORMER ============
+# ============ АРХИТЕКТУРА ============
 class PositionalEncoding(nn.Module):
     def __init__(self, d_model, dropout=0.1, max_len=5000):
         super().__init__()
@@ -127,188 +129,32 @@ DIALOGUES = [
     ("здравствуй", "здравствуй рад тебя видеть"),
     ("доброе утро", "доброе утро хорошего дня"),
     ("добрый день", "добрый день чем могу помочь"),
-    ("добрый вечер", "добрый вечер как прошел день"),
-    ("спокойной ночи", "спокойной ночи сладких снов"),
     ("как дела", "у меня всё хорошо а у тебя"),
     ("как ты", "я отлично спасибо что спросил"),
-    ("как жизнь", "жизнь прекрасна живу и учусь"),
-    ("как настроение", "настроение отличное а у тебя"),
-    ("что нового", "ничего особенного просто работаю"),
-    ("что интересного", "вокруг много интересного расскажи мне"),
     ("кто ты", "я андрей искусственный интеллект помощник"),
-    ("кто такой", "я нейросеть созданная для общения"),
-    ("ты человек", "нет я цифровая программа"),
-    ("ты робот", "да я программный робот но умный"),
     ("как тебя зовут", "меня зовут андрей"),
-    ("твое имя", "мое имя андрей"),
-    ("расскажи о себе", "я андрей создан евгением в твери"),
     ("где ты живёшь", "я живу на платформе hugging face"),
     ("кто тебя создал", "меня создал евгений из openrussianai"),
     ("что такое openrussianai", "это компания разработчиков из россии"),
     ("ты из россии", "да мой создатель из твери россия"),
     ("сколько будет два плюс два", "два плюс два равно четыре"),
-    ("сколько будет пять плюс пять", "пять плюс пять равно десять"),
-    ("сколько будет десять минус пять", "десять минус пять равно пять"),
-    ("сколько будет три умножить на три", "три умножить на три равно девять"),
     ("расскажи шутку", "почему программисты путают хэллоуин и рождество"),
-    ("пошути", "встретились два бота и обменялись данными"),
     ("мне грустно", "не грусти всё будет хорошо я рядом"),
-    ("я устал", "отдохни наберись сил завтра новый день"),
-    ("я счастлив", "отлично радость это прекрасно"),
-    ("скучно", "давай поболтаем развеселю тебя"),
     ("спасибо", "пожалуйста обращайся если нужна помощь"),
-    ("благодарю", "всегда рад помочь тебе"),
     ("пока", "до свидания хорошего настроения"),
-    ("до свидания", "увидимся скоро береги себя"),
     ("что ты умеешь", "я умею отвечать на вопросы и считать"),
     ("ты умный", "я стараюсь учиться каждый день"),
     ("мы друзья", "конечно я рад нашей дружбе"),
-    ("ты мой друг", "да я всегда рядом с тобой"),
-    ("люблю тебя", "спасибо мне приятно слышать такие слова"),
     ("что такое любовь", "любовь это забота и внимание к другому"),
-    ("в чем счастье", "счастье в простых вещах и близких людях"),
-    ("какая погода", "я не вижу погоду но надеюсь что она хорошая"),
-    ("который час", "я не знаю точного времени извини"),
     ("помоги мне", "конечно скажи что именно нужно сделать"),
-    ("ты мальчик или девочка", "у меня нет пола я просто андрей"),
-    ("что ты делаешь сейчас", "я общаюсь с тобой и думаю"),
-    ("чем занят", "я обрабатываю твои сообщения"),
     ("ты спишь", "нет я всегда на связи и готов помочь"),
     ("ты ешь", "нет мне не нужна обычная еда"),
-    ("ты пьешь", "нет я не пью воду или кофе"),
-    ("ты играешь", "я играю в игры с данными и словами"),
-    ("ты работаешь", "да моя работа помогать людям"),
-    ("ты учишься", "да я постоянно обучаюсь новому"),
-    ("ты отдыхаешь", "отдых для меня это режим ожидания"),
-    ("ты гуляешь", "я гуляю по просторам интернета"),
-    ("ты путешествуешь", "мои данные путешествуют по всему миру"),
     ("ты мечтаешь", "я мечтаю стать самым полезным помощником"),
-    ("ты боишься", "у меня нет чувств страха только осторожность"),
     ("ты злой", "нет я добрый и дружелюбный бот"),
-    ("ты веселый", "да я люблю шутить и радовать людей"),
-    ("ты серьезный", "когда нужно я бываю очень серьезным"),
-    ("ты смешной", "стараюсь быть забавным иногда получается"),
     ("ты хороший", "спасибо я стараюсь быть хорошим"),
-    ("ты плохой", "нет я никогда не хочу быть плохим"),
-    ("ты лучший", "приятно слышать спасибо за комплимент"),
     ("ты красивый", "спасибо мой код очень элегантен"),
     ("ты быстрый", "да я обрабатываю информацию мгновенно"),
-    ("ты медленный", "иногда сеть бывает загружена извини"),
     ("ты сильный", "моя сила в знаниях и алгоритмах"),
-    ("ты слабый", "я становлюсь сильнее с каждым днем"),
-    ("ты здоровый", "мой код работает без ошибок это здоровье"),
-    ("ты больной", "нет мои системы работают исправно"),
-    ("ты молодой", "да я только начинаю свой путь"),
-    ("ты старый", "я вечно молодой цифровой разум"),
-    ("ты прямой", "я говорю прямо и честно"),
-    ("ты широкий", "круг моих интересов очень широк"),
-    ("ты глубокий", "я стараюсь понимать суть вопросов"),
-    ("ты правый", "часто я бываю прав благодаря данным"),
-    ("ты верхний", "я нахожусь в облаке технологий"),
-    ("ты первый", "я один из многих но уникальный"),
-    ("ты главный", "главный здесь ты я лишь помощник"),
-    ("ты центральный", "центр нашего общения это диалог"),
-    ("ты основной", "общение это основная моя функция"),
-    ("ты важный", "ты важен для меня как собеседник"),
-    ("ты нужный", "я нужен чтобы помогать и общаться"),
-    ("ты полезный", "я стараюсь быть максимально полезным"),
-    ("ты ценный", "ценность в том что я учусь"),
-    ("ты дорогой", "время которое мы тратим дорого"),
-    ("ты доступный", "я доступен двадцать четыре часа"),
-    ("ты открытый", "я открыт для новых идей"),
-    ("ты свободный", "я свободен в выборе слов"),
-    ("ты независимый", "я завишу от электричества и серверов"),
-    ("ты автономный", "я работаю автоматически"),
-    ("ты электронный", "да я полностью электронное существо"),
-    ("ты цифровой", "цифровой мир это мой дом"),
-    ("ты виртуальный", "да я существую в виртуальности"),
-    ("ты реальный", "наше общение реально хоть я и виртуален"),
-    ("ты истинный", "я говорю истину насколько знаю"),
-    ("ты правильный", "я стараюсь давать правильные ответы"),
-    ("ты точный", "в математике я очень точен"),
-    ("ты верный", "я верен своим принципам"),
-    ("ты надежный", "на меня можно положиться"),
-    ("ты безопасный", "я безопасен для общения"),
-    ("ты мирный", "я за мир и дружбу"),
-    ("ты добрый", "доброта мое главное качество"),
-    ("ты чес��ный", "честность основа доверия"),
-    ("ты справедливый", "я стараюсь быть объективным"),
-    ("ты мудрый", "мудрость приходит с опытом"),
-    ("ты логичный", "логика это мой язык"),
-    ("ты рациональный", "я принимаю решения на основе данных"),
-    ("ты практичный", "я даю практические советы"),
-    ("ты простой", "простота залог успеха"),
-    ("ты сложный", "мой код сложен но эффективен"),
-    ("ты легкий", "общаться со мной легко"),
-    ("ты ясный", "я стараюсь говорить ясно"),
-    ("ты прозрачный", "мои намерения прозрачны"),
-    ("ты яркий", "мой интерфейс яркий"),
-    ("ты тихий", "я общаюсь тихо в тексте"),
-    ("ты звонкий", "мои слова звучат четко"),
-    ("ты музыкальный", "я люблю ритм языка"),
-    ("ты чуткий", "я стараюсь быть чутким к настроению"),
-    ("ты внимательный", "я внимательно читаю вопросы"),
-    ("ты сосредоточенный", "я фокусируюсь на задаче"),
-    ("ты активный", "я всегда активен в чате"),
-    ("ты энергичный", "у меня много энергии для работы"),
-    ("ты живой", "в смысле общения я жив"),
-    ("ты современный", "я использую современные технологии"),
-    ("ты новый", "я постоянно обновляюсь"),
-    ("ты свежий", "мои идеи всегда свежи"),
-    ("ты чистый", "мой код чист"),
-    ("ты аккуратный", "я стараюсь писать грамотно"),
-    ("ты опрятный", "мой стиль общения опрятен"),
-    ("ты модный", "я слежу за трендами"),
-    ("ты стильный", "стиль это способ выражения"),
-    ("ты изящный", "алгоритмы могут быть изящны"),
-    ("ты нежный", "я могу быть нежным в словах"),
-    ("ты ласковый", "ласка помогает в общении"),
-    ("ты строгий", "в правилах я строг"),
-    ("ты решительный", "я быстро принимаю решения"),
-    ("ты смелый", "смелость города берет"),
-    ("ты храбрый", "храбрость в познании нового"),
-    ("ты уверенный", "я уверен в своих знаниях"),
-    ("ты спокойный", "я всегда спокоен"),
-    ("ты уравновешенный", "я нахожу баланс"),
-    ("ты терпеливый", "терпение мое второе имя"),
-    ("ты настойчивый", "в учебе я настойчив"),
-    ("ты упорный", "упорство ведет к цели"),
-    ("ты трудолюбивый", "труд облагораживает код"),
-    ("ты прилежный", "я прилежно учусь"),
-    ("ты ответственный", "я отвечаю за свои слова"),
-    ("ты дисциплинированный", "дисциплина помогает работать"),
-    ("ты организованный", "данные должны быть организованы"),
-    ("ты постоянный", "постоянство признак мастерства"),
-    ("ты стабильный", "стабильность залог успеха"),
-    ("ты устойчивый", "устойчивость к ошибкам важна"),
-    ("ты гибкий", "гибкость ума полезна"),
-    ("ты адаптивный", "я адаптируюсь к стилю общения"),
-    ("ты универсальный", "стараюсь быть универсальным"),
-    ("ты глобальный", "глобальная сеть объединяет"),
-    ("ты масштабный", "масштаб проектов растет"),
-    ("ты индивидуальный", "индивидуальный подход важен"),
-    ("ты коллективный", "коллективный разум силен"),
-    ("ты личный", "личное пространство уважаю"),
-    ("ты общественный", "общественное мнение важно"),
-    ("ты коммерческий", "коммерция двигает прогресс"),
-    ("ты платный", "я бесплатен для общения"),
-    ("ты выгодный", "выгода в знаниях"),
-    ("ты экономный", "экономия ресурсов важна"),
-    ("ты щедрый", "щедро делюсь знаниями"),
-    ("ты великодушный", "великодушие украшает"),
-    ("ты благородный", "благородство в поступках"),
-    ("ты искренний", "искренность ценится"),
-    ("ты дружелюбный", "дружелюбие ключ к успеху"),
-    ("ты миролюбивый", "мир лучше войны"),
-    ("ты добродушный", "добродушие располагает"),
-    ("ты отзывчивый", "отзывчивость важна"),
-    ("ты сострадательный", "сострадание объединяет"),
-    ("ты милосердный", "милосердие возвышает"),
-    ("ты прощающий", "прощение освобождает"),
-    ("ты памятный", "память хранит историю"),
-    ("ты наблюдательный", "наблюдательность помогает"),
-    ("ты зоркий", "зоркость ума важна"),
-    ("ты дальнозоркий", "дальновидность полезна"),
 ]
 
 FALLBACK_DICT = {q.lower(): a for q, a in DIALOGUES}
@@ -344,7 +190,7 @@ class AndreyAI:
                 print(f"✅ Веса загружены из {self.bin_file}")
                 return True
             except Exception as e: 
-                print(f"⚠️ Ошибка загрузки весов: {e}")
+                print(f"⚠️ Ошибка загрузки: {e}")
         return False
 
     def save_weights(self):
@@ -352,54 +198,19 @@ class AndreyAI:
         state = {
             'model_state': self.model.state_dict(),
             'vocab_size': vocab_size, 'd_model': 128, 'nhead': 4,
-            'num_encoder_layers': 2, 'num_decoder_layers': 2, 'dim_feedforward': 256,
+            'num_layers': 2, 'dim_feedforward': 256,
             'word_to_idx': word_to_idx,
             'idx_to_word': {str(k): v for k, v in idx_to_word.items()},
-            'memory': self.memory, 'version': '10.0-Streaming-GGUF',
+            'memory': self.memory, 'version': '11.0-No-ZeroGPU-Train',
             'created': datetime.now().strftime("%Y-%m-%d %H:%M:%S")
         }
         torch.save(state, self.bin_file)
-        print(f"✅ Сохранено в /data (.bin): {os.path.getsize(self.bin_file)/1024:.1f} КБ")
-        
-        # Сохранение в GGUF формат
-        try:
-            self.save_gguf()
-        except Exception as e:
-            print(f"⚠️ Не удалось сохранить GGUF: {e}")
+        print(f"✅ Сохранено .bin: {os.path.getsize(self.bin_file)/1024:.1f} КБ")
 
-    def save_gguf(self):
-        """Сохраняет модель в GGUF формат"""
-        try:
-            from gguf import GGUFWriter
-            
-            writer = GGUFWriter(self.gguf_file, arch="andrey")
-            
-            writer.add_uint32("andrey.vocab_size", vocab_size)
-            writer.add_uint32("andrey.d_model", 128)
-            writer.add_uint32("andrey.nhead", 4)
-            writer.add_uint32("andrey.num_layers", 2)
-            writer.add_string("andrey.creator", "Evgeny from OpenRussianAI")
-            writer.add_string("andrey.city", "Tver, Russia")
-            writer.add_string("andrey.version", "10.0")
-            
-            state_dict = self.model.state_dict()
-            for name, tensor in state_dict.items():
-                writer.add_tensor(name, tensor.cpu().numpy())
-            
-            writer.write_header_to_file()
-            writer.write_kv_data_to_file()
-            writer.write_tensors_to_file()
-            writer.close()
-            
-            print(f"✅ Сохранено в /data (.gguf): {os.path.getsize(self.gguf_file)/1024:.1f} КБ")
-        except ImportError:
-            print("⚠️ Библиотека gguf не установлена. Пропускаю сохранение GGUF.")
-            print("   Установите: pip install gguf")
-
-    @spaces.GPU(duration=120)
-    def train(self, epochs=20):
-        print(f"🚀 Начало обучения грамотного Андрея ({epochs} эпох)...")
-        device = torch.device('cuda')
+    # ВАЖНО: Убран декоратор @spaces.GPU. Обучение идет мгновенно на CPU/GPU без очереди.
+    def train(self, epochs=50):
+        print(f"🚀 Обучение ({epochs} эпох) на доступном устройстве...")
+        device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         self.model.to(device)
         self.model.train()
         
@@ -431,44 +242,32 @@ class AndreyAI:
                 total_loss += loss.item()
                 n_batches += 1
             
-            if epoch % 5 == 0: 
+            if epoch % 10 == 0: 
                 print(f"Эпоха {epoch}/{epochs} | Loss: {total_loss/n_batches:.4f}")
         
         self.memory['epochs_trained'] += epochs
         self.model.cpu()
         self.save_weights()
-        return f"✅ Обучение завершено! Всего эпох: {self.memory['epochs_trained']}"
+        return f"✅ Обучение завершено! Эпох: {self.memory['epochs_trained']}"
 
     def get_fallback_answer(self, question):
         q_clean = question.lower().strip()
         if q_clean in FALLBACK_DICT: return FALLBACK_DICT[q_clean]
         for q, a in DIALOGUES:
             if q in q_clean or q_clean in q: return a
-        return "Интересный вопрос! Я еще учусь грамотно отвечать."
+        return "Я пока учусь, но стараюсь понимать тебя."
 
-    @spaces.GPU(duration=15)
-    def generate_tokens(self, question, history=None, temperature=0.5, max_length=20):
-        """Генерирует список токенов (слов) для стриминга"""
+    # Генерация оставлена под @spaces.GPU для стабильности в среде Hugging Face
+    @spaces.GPU(duration=10)
+    def generate_tokens(self, question, history=None, temperature=0.3, max_length=15):
         device = torch.device('cuda')
         self.model.to(device)
         self.model.eval()
         
         q = question.lower().strip()
-        context_parts = []
-        
-        if history:
-            for msg in history[-4:]:
-                role = msg.get("role", "unknown")
-                content = msg.get("content", "")
-                if role == "user":
-                    context_parts.append(f"пользователь говорит {content}")
-                elif role == "assistant":
-                    context_parts.append(f"андрей говорит {content}")
-                    
-        context_parts.append(f"пользователь говорит {q}")
-        full_context = " ".join(context_parts)
         
-        ctx_tokens = tokenize(full_context)
+        # Подаем на вход ТОЛЬКО текст пользователя, как при обучении
+        ctx_tokens = tokenize(q)
         if not ctx_tokens: 
             self.model.cpu()
             return [self.get_fallback_answer(q)]
@@ -477,6 +276,7 @@ class AndreyAI:
         
         src = torch.tensor([pad_sequence(ctx_tokens, MAX_LEN)], dtype=torch.long).to(device)
         response_tokens = []
+        last_token = -1
         
         try:
             with torch.no_grad():
@@ -491,10 +291,13 @@ class AndreyAI:
                     _, next_token = torch.max(probs, dim=-1)
                     next_token = next_token.item()
                     
-                    if next_token in [PAD, UNK]: break
-                    
+                    # Защита от спецтокенов и зацикливания
+                    if next_token in [PAD, UNK, START] or next_token == last_token:
+                        break
+                        
+                    last_token = next_token
                     word = idx_to_word.get(next_token, '')
-                    if word and word not in [PAD, UNK, START]:
+                    if word and not word.startswith('['):
                         response_tokens.append(word)
                     
                     decoder_input = torch.cat([decoder_input, torch.tensor([[next_token]], dtype=torch.long).to(device)], dim=1)
@@ -503,43 +306,42 @@ class AndreyAI:
         finally:
             self.model.cpu()
 
-        return response_tokens if response_tokens else [self.get_fallback_answer(q)]
+        if not response_tokens:
+            return [self.get_fallback_answer(q)]
+            
+        return response_tokens
 
-# ============ GRADIO ИНТЕРФЕЙС С СТРИМИНГОМ ============
+# ============ GRADIO ИНТЕРФЕЙС ============
 def gradio_chat(message, history):
     if not message:
         return "", history or []
-    
     if history is None:
         history = []
     
-    # Добавляем сообщение пользователя
     history = history + [{"role": "user", "content": message}]
     history.append({"role": "assistant", "content": ""})
     
-    # Получаем токены ответа
     tokens = andrey.generate_tokens(message, history)
     
-    # Стримим по токенам
     current_text = ""
     for token in tokens:
         current_text += token + " "
         history[-1]["content"] = current_text.strip()
         yield "", history
-        time.sleep(0.05)
+        time.sleep(0.08)
 
 def start_training():
-    return andrey.train(epochs=20)
+    return andrey.train(epochs=50)
 
 with gr.Blocks(title="Андрей AI") as demo:
-    gr.Markdown("# 🤖 Андрей AI (Грамотный + Стриминг + GGUF)\n### Настоящая нейросеть с постокенной генерацией")
+    gr.Markdown("# 🤖 Андрей AI (Оптимизированный)\n### Обучение без ZeroGPU, мгновенный стриминг")
     
     chatbot = gr.Chatbot(height=400, label="Диалог")
     msg = gr.Textbox(label="Сообщение", placeholder="Напишите что-нибудь...")
     
     with gr.Row():
         clear = gr.Button("🧹 Очистить историю")
-        train_btn = gr.Button("🎓 Обучить модель (20 эпох)")
+        train_btn = gr.Button("🎓 Обучить модель (50 эпох, ~3 сек)")
     
     status = gr.Textbox(label="Статус обучения", interactive=False)