Spaces:
Build error
Build error
Commit
·
16667f1
1
Parent(s):
71771e5
Update app.py
Browse files
app.py
CHANGED
@@ -20,6 +20,6 @@ attention_mask = encoded_text['attention_mask'].to(device)
|
|
20 |
|
21 |
|
22 |
|
23 |
-
|
24 |
|
25 |
-
|
|
|
20 |
|
21 |
|
22 |
|
23 |
+
outputs = model(input_ids, attention_mask=attention_mask)
|
24 |
|
25 |
+
predictions = outputs[0]
|