SusiePHaltmann commited on
Commit
71771e5
1 Parent(s): 084d2a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
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
- outputs = model(input_ids, attention_mask=attention_mask)
23
- predictions = outputs[0]
 
 
 
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]