Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,10 +1,13 @@
|
|
| 1 |
-
#
|
|
|
|
| 2 |
import gradio as gr
|
| 3 |
from transformers import pipeline
|
| 4 |
|
| 5 |
-
# Загружаем
|
| 6 |
-
sentiment_pipe = pipeline(
|
| 7 |
-
|
|
|
|
|
|
|
| 8 |
|
| 9 |
def analyze_sentiment(text: str) -> str:
|
| 10 |
text = text.strip()
|
|
@@ -15,31 +18,28 @@ def analyze_sentiment(text: str) -> str:
|
|
| 15 |
label = result["label"]
|
| 16 |
score = result["score"]
|
| 17 |
|
| 18 |
-
#
|
| 19 |
-
if label
|
| 20 |
-
label_ru = "НЕГАТИВНАЯ ТОНАЛЬНОСТЬ"
|
| 21 |
-
elif label.upper().startswith("POS"):
|
| 22 |
label_ru = "ПОЗИТИВНАЯ ТОНАЛЬНОСТЬ"
|
|
|
|
|
|
|
| 23 |
else:
|
| 24 |
-
label_ru =
|
| 25 |
|
| 26 |
return f"{label_ru} (уверенность модели: {score:.2f})"
|
| 27 |
|
| 28 |
|
| 29 |
-
#
|
| 30 |
demo = gr.Interface(
|
| 31 |
fn=analyze_sentiment,
|
| 32 |
inputs=gr.Textbox(
|
| 33 |
lines=5,
|
| 34 |
-
label="Введите текст (
|
| 35 |
-
placeholder="Например: Мне нравится этот
|
| 36 |
),
|
| 37 |
outputs=gr.Textbox(label="Результат анализа"),
|
| 38 |
-
title="Sentiment Demo",
|
| 39 |
-
description=(
|
| 40 |
-
"Пример простого приложения.\n"
|
| 41 |
-
"Модель определяет тональность текста."
|
| 42 |
-
),
|
| 43 |
)
|
| 44 |
|
| 45 |
if __name__ == "__main__":
|
|
|
|
| 1 |
+
# ПРОСТОЙ МНОГОЛИНГВАЛЬНЫЙ АНАЛИЗ ТОНАЛЬНОСТИ (RU + EN + UZ)
|
| 2 |
+
|
| 3 |
import gradio as gr
|
| 4 |
from transformers import pipeline
|
| 5 |
|
| 6 |
+
# Загружаем многоязычную модель (поддерживает русский, английский и узбекский)
|
| 7 |
+
sentiment_pipe = pipeline(
|
| 8 |
+
"sentiment-analysis",
|
| 9 |
+
model="cardiffnlp/twitter-xlm-roberta-base-sentiment"
|
| 10 |
+
)
|
| 11 |
|
| 12 |
def analyze_sentiment(text: str) -> str:
|
| 13 |
text = text.strip()
|
|
|
|
| 18 |
label = result["label"]
|
| 19 |
score = result["score"]
|
| 20 |
|
| 21 |
+
# Интерпретация меток модели
|
| 22 |
+
if label == "positive":
|
|
|
|
|
|
|
| 23 |
label_ru = "ПОЗИТИВНАЯ ТОНАЛЬНОСТЬ"
|
| 24 |
+
elif label == "negative":
|
| 25 |
+
label_ru = "НЕГАТИВНАЯ ТОНАЛЬНОСТЬ"
|
| 26 |
else:
|
| 27 |
+
label_ru = "НЕЙТРАЛЬНАЯ ТОНАЛЬНОСТЬ"
|
| 28 |
|
| 29 |
return f"{label_ru} (уверенность модели: {score:.2f})"
|
| 30 |
|
| 31 |
|
| 32 |
+
# Gradio UI
|
| 33 |
demo = gr.Interface(
|
| 34 |
fn=analyze_sentiment,
|
| 35 |
inputs=gr.Textbox(
|
| 36 |
lines=5,
|
| 37 |
+
label="Введите текст (Русский / O'zbekcha / English)",
|
| 38 |
+
placeholder="Например: Мне нравится этот продукт!\nMasalan: Menga bu mahsulot yoqadi!\nExample: I like this product!"
|
| 39 |
),
|
| 40 |
outputs=gr.Textbox(label="Результат анализа"),
|
| 41 |
+
title="Multilingual Sentiment Demo",
|
| 42 |
+
description="Многоязычный анализ тональности текста (RU + UZ + EN)."
|
|
|
|
|
|
|
|
|
|
| 43 |
)
|
| 44 |
|
| 45 |
if __name__ == "__main__":
|