Spaces:
Build error
Build error
Commit
·
084d2a3
1
Parent(s):
78e9acd
Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,5 @@ input_ids = encoded_text['input_ids'].to(device)
|
|
19 |
attention_mask = encoded_text['attention_mask'].to(device)
|
20 |
|
21 |
|
22 |
-
|
23 |
-
|
24 |
-
|
25 |
-
predictions = outputs[0]
|
|
|
19 |
attention_mask = encoded_text['attention_mask'].to(device)
|
20 |
|
21 |
|
22 |
+
outputs = model(input_ids, attention_mask=attention_mask)
|
23 |
+
predictions = outputs[0]
|
|
|
|