Helder Rodrigues commited on
Commit
082c008
1 Parent(s): 1645f2b

Add application file

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ classifier = pipeline("sentiment-analysis")
5
 
6
  def generate(text):
7
  result = classifier(text)
8
- return result[0].label
9
 
10
  examples = [
11
  ["SE is the best company in the world!"],
 
5
 
6
  def generate(text):
7
  result = classifier(text)
8
+ return result[0]['label']
9
 
10
  examples = [
11
  ["SE is the best company in the world!"],