Update app.py

app.py

@@ -1,46 +1,51 @@
-# ПРОСТЕЙШИЙ ПРИМЕР АНАЛИЗА ТОНАЛЬНОСТИ ТЕКСТА
+# Простейший пример анализа тональности текста
 import gradio as gr
 from transformers import pipeline
 
-# Загружаем готовый пайплайн из Hugging Face
-sentiment_pipe = pipeline("sentiment-analysis",
-                          model="blanchefort/rubert-base-cased-sentiment")
+# Загружаем готовый pipeline с Hugging Face
+translation_pipe = pipeline("translation_ru_to_uz", model="sarahai/nllb-ru-uz")
+
+sentiment_pipe = pipeline(
+    "sentiment-analysis",
+    model="blanchefort/rubert-base-cased-sentiment"
+)
 
 def analyze_sentiment(text: str) -> str:
     text = text.strip()
     if not text:
         return "Введите текст для анализа."
 
-    result = sentiment_pipe(text)[0]
+translation = translator_pipe(text)
+
+    result = sentiment_pipe(translation)[0]
     label = result["label"]
     score = result["score"]
 
-    # Определяем тональность
+    # Перевод меток модели
     if label.upper().startswith("NEG"):
-        label_ru = "НЕГАТИВНАЯ ТОНАЛЬНОСТЬ"
+        label_ru = "Негативная тональность"
     elif label.upper().startswith("POS"):
-        label_ru = "ПОЗИТИВНАЯ ТОНАЛЬНОСТЬ"
+        label_ru = "Позитивная тональность"
     else:
-        label_ru = f"ТОНАЛЬНОСТЬ: {label}"
+        label_ru = f"Тональность: {label}"
 
     return f"{label_ru} (уверенность модели: {score:.2f})"
 
-
 # Описание интерфейса
 demo = gr.Interface(
     fn=analyze_sentiment,
     inputs=gr.Textbox(
         lines=5,
-        label="Введите текст (поддерживается русский и английский)",
-        placeholder="Например: Мне нравится этот продукт!",
+        label="Введите текст (желательно на русском)",
+        placeholder="Например: Мне нравится этот продукт!"
     ),
     outputs=gr.Textbox(label="Результат анализа"),
     title="Sentiment Demo",
     description=(
-        "Пример простого приложения.\n"
+        "Пример простого приложения на Hugging Face.\n"
         "Модель определяет тональность текста."
     ),
 )
 
-if __name__ == "__main__":
+if name == "main":
     demo.launch()