dahongj commited on
Commit
123b344
1 Parent(s): c45446a
Files changed (1) hide show
  1. app.py +9 -2
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
- print(specific_model(data))
 
 
 
 
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():