Korsholm22 commited on
Commit
25d1623
·
1 Parent(s): 2865f55

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
6
  #Defining the classify function which takes text as input and returns the label of the sentiment
7
  def classify(text):
8
  # Initializing the pipeline for sentiment analysis
9
- cls = pipeline('text-classification', model='Korsholm22/dk_emotion_bert_class')
10
  # Predicting the sentiment label for the input text
11
  return cls(text)[0]['label']
12
 
 
6
  #Defining the classify function which takes text as input and returns the label of the sentiment
7
  def classify(text):
8
  # Initializing the pipeline for sentiment analysis
9
+ cls = pipeline('text-classification', model='RJuro/dk_emotion_bert_in_class')
10
  # Predicting the sentiment label for the input text
11
  return cls(text)[0]['label']
12