Xaaddad commited on
Commit
39aba33
·
verified ·
1 Parent(s): c1dde7b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -17
app.py CHANGED
@@ -1,10 +1,13 @@
1
- # ПРОСТЕЙШИЙ ПРИМЕР АНАЛИЗА ТОНАЛЬНОСТИ ТЕКСТА
 
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()
@@ -15,31 +18,28 @@ def analyze_sentiment(text: str) -> str:
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__":
 
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__":