Fawaz commited on
Commit
ae4f84a
β€’
1 Parent(s): 217433a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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/'