Martijn van Beers
commited on
Commit
•
ef158cd
1
Parent(s):
4b1b415
fix typo
Browse files
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
|
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]
|