ReiderMx commited on
Commit
1a25b0f
verified
1 Parent(s): c8faabf

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +38 -0
app.py ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import pickle
3
+ from transformers import pipeline
4
+
5
+ def load_model(selected_model):
6
+ with open(selected_model, 'rb') as file:
7
+ loaded_model = pickle.load(file)
8
+ return loaded_model
9
+
10
+ encoder = {
11
+ 'negative':'assets/negative.jpg',
12
+ 'neutral':'assets/neutral.jpg',
13
+ 'positive':'assets/positive.jpeg'
14
+ }
15
+
16
+ classifier = pipeline(task="zero-shot-classification", model="facebook/bart-large-mnli")
17
+ def analyze_sentiment(text):
18
+ results = classifier(text,["positive","negative",'neutral'],multi_label=True)
19
+ mx = max(results['scores'])
20
+ ind = results['scores'].index(mx)
21
+ result = results['labels'][ind]
22
+ return encoder[result]
23
+
24
+ description = """
25
+ <p>Este experimento tiene como objetivo evaluar la capacidad de un modelo de inteligencia artificial para comprender los sentimientos expresados en una frase. Este an谩lisis puede ser 煤til para determinar el porcentaje de sentimientos positivos, negativos o neutrales en los comentarios de una publicaci贸n.</p>
26
+ <br>
27
+ <p>Proyecto realizado por alumnos de UPN</p>
28
+ <br>
29
+ <p>--Sistemas Inteligentes y Machine Learning--</p>
30
+ """
31
+
32
+
33
+ iface = gr.Interface(fn=analyze_sentiment,
34
+ title="Analisis de Sentimientos",
35
+ description = description,
36
+ inputs="text",
37
+ outputs="image"
38
+ iface.launch(share=True)