Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -146,6 +146,7 @@ def get_inputs(tokenizer):
|
|
146 |
segment_ids = torch.tensor(segment_ids, dtype=torch.long)
|
147 |
|
148 |
return input_ids.unsqueeze(0).to(device), segment_ids.unsqueeze(0).to(device)
|
|
|
149 |
|
150 |
img_size = 224
|
151 |
ckpt_path = 'ACTX_p/'
|
|
|
146 |
segment_ids = torch.tensor(segment_ids, dtype=torch.long)
|
147 |
|
148 |
return input_ids.unsqueeze(0).to(device), segment_ids.unsqueeze(0).to(device)
|
149 |
+
|
150 |
|
151 |
img_size = 224
|
152 |
ckpt_path = 'ACTX_p/'
|