pzangara commited on
Commit
94236da
1 Parent(s): 1994179

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -10,6 +10,10 @@ import pandas as pd
10
  model_name = "MoritzLaurer/mDeBERTa-v3-base-xnli-multilingual-nli-2mil7"
11
  tokenizer = AutoTokenizer.from_pretrained(model_name)
12
 
 
 
 
 
13
 
14
  def classify_text(text):
15
  """
@@ -19,9 +23,8 @@ def classify_text(text):
19
  Returns:
20
  str: La clasificación del texto, que puede ser "Positivo", "Negativo" o "Neutro".
21
  """
22
- analyzer = create_analyzer(task="sentiment", lang="es")
23
- result = analyzer.predict(text)[0]['label']
24
- #result = classifier(text)[0]['label']
25
  if result == "POS":
26
  return "Positivo"
27
  elif result == "NEG":
@@ -37,6 +40,8 @@ def clasificador(input1, input2):
37
  output0 = classifier(sequence_to_classify, candidate_labels, multi_label=False)
38
  output1=pd.DataFrame(output0)
39
  output1=output1.iloc[:,1:3]
 
 
40
  output2=classify_text(input1)
41
  return output1, output2
42
 
 
10
  model_name = "MoritzLaurer/mDeBERTa-v3-base-xnli-multilingual-nli-2mil7"
11
  tokenizer = AutoTokenizer.from_pretrained(model_name)
12
 
13
+ # Inicializa Modelo de Clasificación de Sentimientos
14
+ model_name = 'pysentimiento/robertuito-sentiment-analysis'
15
+ classifier_sent = pipeline("text-classification", model=model_name)
16
+
17
 
18
  def classify_text(text):
19
  """
 
23
  Returns:
24
  str: La clasificación del texto, que puede ser "Positivo", "Negativo" o "Neutro".
25
  """
26
+
27
+ result = classifier_sent(text)[0]['label']
 
28
  if result == "POS":
29
  return "Positivo"
30
  elif result == "NEG":
 
40
  output0 = classifier(sequence_to_classify, candidate_labels, multi_label=False)
41
  output1=pd.DataFrame(output0)
42
  output1=output1.iloc[:,1:3]
43
+ #analyzer = create_analyzer(task="sentiment", lang="es")
44
+ #output2 = analyzer.predict(input1)
45
  output2=classify_text(input1)
46
  return output1, output2
47