Martijn van Beers commited on
Commit
ef158cd
1 Parent(s): 4b1b415
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -202,7 +202,7 @@ def sentence_sentiment(input_text):
202
  attention_mask = encoding["attention_mask"].to(device)
203
  output = model(input_ids=input_ids, attention_mask=attention_mask)[0]
204
  index = output.argmax(axis=-1).item()
205
- return classification[index]
206
 
207
  def sentiment_explanation_hila(input_text):
208
  text_batch = [input_text]
 
202
  attention_mask = encoding["attention_mask"].to(device)
203
  output = model(input_ids=input_ids, attention_mask=attention_mask)[0]
204
  index = output.argmax(axis=-1).item()
205
+ return classifications[index]
206
 
207
  def sentiment_explanation_hila(input_text):
208
  text_batch = [input_text]