Spaces:
Build error
Build error
SusiePHaltmann
commited on
Commit
•
71771e5
1
Parent(s):
084d2a3
Update app.py
Browse files
app.py
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
import torch
|
2 |
from transformers import *
|
3 |
|
4 |
# Choose the device and load the model
|
@@ -19,5 +19,7 @@ input_ids = encoded_text['input_ids'].to(device)
|
|
19 |
attention_mask = encoded_text['attention_mask'].to(device)
|
20 |
|
21 |
|
22 |
-
|
23 |
-
|
|
|
|
|
|
1 |
+
import torch
|
2 |
from transformers import *
|
3 |
|
4 |
# Choose the device and load the model
|
|
|
19 |
attention_mask = encoded_text['attention_mask'].to(device)
|
20 |
|
21 |
|
22 |
+
|
23 |
+
outputs = model(input_ids, attention_mask=attention_mask)
|
24 |
+
|
25 |
+
predictions = outputs[0]
|