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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +38 -30
app.py CHANGED
@@ -2,37 +2,45 @@
2
  import gradio as gr
3
  from transformers import pipeline
4
 
5
- #ЗАГРУЖАЕМ ГОТОВЫЙ ПАЙПЛАЙН ИЗ ХАГЕН ФАСЕ
6
- sentiment_pipe=pipeline("Sentiment-analysis", model="blanchefort/rubert-base-cased-sentiment")
7
- def analyze_sentiment(text: str)->str:
8
- text= text.strip()
9
- if not text:
10
- return "ВВЕЕДИТЕ ТЕКСТ ДЛЯ АНААЛИЗА"
11
- result=sentiment_pipe(text)[0]
12
- label=result["label"]
13
- label=result["score"]
14
-
15
- if label.uppper().startsvith("NEG"):
16
- label_ru="НЕГАТИВНАЯ ТОНАЛЬНОСТЬ"
17
- elif lable.uppper().startsvith("POS"):
18
- label_ru="ПОЗИТИВНАЯ ТОНАЛЬНОСТЬ"
19
- else : label_ru=f"ТОНАЛЬНОСТЬ:{label}"
20
- return f"{lable_ru} (УВЕРЕННОСТЬ МОДЕЛИ {score: .2f})"
21
-
22
- #ОПИСАНИЕ ИНТЕРФЕЙСА
23
- demo=gr.Interface(
 
 
 
 
 
 
 
24
  fn=analyze_sentiment,
25
- input=gr.Textbox(
26
- lines=5,
27
- label="ВВЕДИТЕ ТЕКСТ (ЖЕЛАТЕЛЬНО НА АНГЛИЙСКОМ ЯЗЫКЕ)",
28
- placeholder="НА ПРИМЕР: I like this product!",
29
  ),
30
- output=gr.Textbox(label="РЕЗУЛЬТАТ АНАЛИЗА"),
31
- title="Sentiment demo",
32
  description=(
33
- "ПРИМЕР ПРОСТОГО ПРИЛОЖЕНИЯ.\n"
34
- "МОДЕЛЬ ОПРЕДЕЛЯТ ТОНАЛЬНОСТЬ ТЕКСТА",
35
  ),
36
- )
37
- if __name__=="__main__":
38
- demo.launch()
 
 
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()