Update app.py

app.py

@@ -1,15 +1,18 @@
 import gradio as gr
 from transformers import pipeline, AutoTokenizer
-import threading, time
+import threading
+import warnings
 from langdetect import detect
 
+warnings.filterwarnings("ignore")
+
 models = {}
 models_ready = False
+init_message = "⏳ Инициализация моделей (1–2 мин)..."
 
 def load_model_safe(model_name, task="summarization"):
     try:
         print(f"🔄 Загружается модель: {model_name}")
-        # Принудительно используем legacy=False только если доступен fast tokenizer
         tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False, legacy=True)
         model = pipeline(task, model=model_name, tokenizer=tokenizer)
         return model
@@ -18,16 +21,16 @@ def load_model_safe(model_name, task="summarization"):
         return None
 
 def preload_models():
-    global models, models_ready
+    global models, models_ready, init_message
     try:
-        print("🚀 Инициализация моделей...")
         models["en"] = load_model_safe("facebook/bart-large-cnn")
         models["ru"] = load_model_safe("IlyaGusev/mbart_ru_sum_gazeta")
         models["multi"] = load_model_safe("csebuetnlp/mT5_multilingual_XLSum")
         models_ready = True
-        print("✅ Все модели успешно загружены.")
+        init_message = "✅ Все модели успешно загружены!"
+        print(init_message)
     except Exception as e:
-        print(f"❌ Ошибка при инициализации моделей: {e}")
+        init_message = f"❌ Ошибка инициализации моделей: {e}"
 
 threading.Thread(target=preload_models, daemon=True).start()
 
@@ -35,7 +38,7 @@ def summarize_text(text):
     if not text.strip():
         return "⚠️ Пожалуйста, введите текст."
     if not models_ready:
-        return "⏳ Модели ещё загружаются, попробуйте через несколько секунд."
+        return init_message
     try:
         lang = detect(text)
         if lang == "ru":
@@ -49,11 +52,12 @@ def summarize_text(text):
     except Exception as e:
         return f"⚠️ Ошибка обработки: {e}"
 
-iface = gr.Interface(
-    fn=summarize_text,
-    inputs=gr.Textbox(label="Введите текст для суммаризации", lines=12),
-    outputs=gr.Textbox(label="Результат суммаризации"),
-    title="🧠 Eroha Summarizer PRO",
-    description="Многоязычный суммаризатор (EN/RU/Multilingual) на основе BART, MBART, mT5.",
-)
+with gr.Blocks(title="Eroha Summarizer PRO") as iface:
+    gr.Markdown("## 🧠 **Eroha Summarizer PRO**\nМногоязычный суммаризатор на BART / MBART / mT5\n")
+    gr.Markdown(init_message)
+    inp = gr.Textbox(label="Введите текст для суммаризации", lines=12)
+    out = gr.Textbox(label="Результат суммаризации")
+    btn = gr.Button("🚀 Суммаризировать")
+    btn.click(summarize_text, inputs=inp, outputs=out)
+
 iface.launch(server_name="0.0.0.0", server_port=7860)