Spaces:
Runtime error
Runtime error
edit
Browse files
app.py
CHANGED
@@ -5,12 +5,19 @@ import torch
|
|
5 |
|
6 |
def bertweet(data):
|
7 |
specific_model = pipeline(model="finiteautomata/bertweet-base-sentiment-analysis")
|
8 |
-
|
|
|
|
|
|
|
|
|
9 |
|
10 |
|
11 |
def getSent(data, model):
|
12 |
if(model == 'Bertweet'):
|
13 |
-
bertweet(data)
|
|
|
|
|
|
|
14 |
|
15 |
|
16 |
def rendPage():
|
|
|
5 |
|
6 |
def bertweet(data):
|
7 |
specific_model = pipeline(model="finiteautomata/bertweet-base-sentiment-analysis")
|
8 |
+
result = specific_model(data)
|
9 |
+
label = result['label']
|
10 |
+
score = result['score']
|
11 |
+
|
12 |
+
return label, score
|
13 |
|
14 |
|
15 |
def getSent(data, model):
|
16 |
if(model == 'Bertweet'):
|
17 |
+
label,score = bertweet(data)
|
18 |
+
col1, col2 = st.columns(2)
|
19 |
+
col1.metric("Feeling",label,None)
|
20 |
+
col2.metric("Score",score,None)
|
21 |
|
22 |
|
23 |
def rendPage():
|