Sasidhar commited on
Commit
e7f3263
·
1 Parent(s): 750105f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ def get_sentiment_model():
24
  def get_sentiment(text):
25
  input_ids = sentiment_tokenizer .encode(text + '</s>', return_tensors='pt')
26
  output = sentiment_model.generate(input_ids=input_ids,max_length=2)
27
- dec = [sentiemnt_tokenizer.decode(ids) for ids in output]
28
  label = dec[0]
29
  return label
30
 
 
24
  def get_sentiment(text):
25
  input_ids = sentiment_tokenizer .encode(text + '</s>', return_tensors='pt')
26
  output = sentiment_model.generate(input_ids=input_ids,max_length=2)
27
+ dec = [sentiment_tokenizer.decode(ids) for ids in output]
28
  label = dec[0]
29
  return label
30