Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,37 +2,45 @@
|
|
| 2 |
import gradio as gr
|
| 3 |
from transformers import pipeline
|
| 4 |
|
| 5 |
-
|
| 6 |
-
sentiment_pipe=pipeline("
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
|
| 10 |
-
|
| 11 |
-
|
| 12 |
-
|
| 13 |
-
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
fn=analyze_sentiment,
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
|
| 29 |
),
|
| 30 |
-
|
| 31 |
-
title="Sentiment
|
| 32 |
description=(
|
| 33 |
-
|
| 34 |
-
|
| 35 |
),
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
|
|
|
|
|
| 2 |
import gradio as gr
|
| 3 |
from transformers import pipeline
|
| 4 |
|
| 5 |
+
# Загружаем готовый пайплайн из Hugging Face
|
| 6 |
+
sentiment_pipe = pipeline("sentiment-analysis",
|
| 7 |
+
model="blanchefort/rubert-base-cased-sentiment")
|
| 8 |
+
|
| 9 |
+
def analyze_sentiment(text: str) -> str:
|
| 10 |
+
text = text.strip()
|
| 11 |
+
if not text:
|
| 12 |
+
return "Введите текст для анализа."
|
| 13 |
+
|
| 14 |
+
result = sentiment_pipe(text)[0]
|
| 15 |
+
label = result["label"]
|
| 16 |
+
score = result["score"]
|
| 17 |
+
|
| 18 |
+
# Определяем тональность
|
| 19 |
+
if label.upper().startswith("NEG"):
|
| 20 |
+
label_ru = "НЕГАТИВНАЯ ТОНАЛЬНОСТЬ"
|
| 21 |
+
elif label.upper().startswith("POS"):
|
| 22 |
+
label_ru = "ПОЗИТИВНАЯ ТОНАЛЬНОСТЬ"
|
| 23 |
+
else:
|
| 24 |
+
label_ru = f"ТОНАЛЬНОСТЬ: {label}"
|
| 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__":
|
| 46 |
+
demo.launch()
|