Tensor loading

app.py

@@ -131,13 +131,17 @@ def evaluate_jaccard(text, selected_text, acronym, offsets, idx_start, idx_end):
 def disambiguate(text, acronym):
     
     inputs = process_data(text, acronym, acronym, TOKENIZER, MAX_LEN)
-    ids = torch.tensor(input['ids']).view(1, -1)
-    mask = torch.tensor(inputs['mask']).view(1, -1)
-    token_type = torch.tensor(inputs['token_type']).view(1, -1)
+    ids = torch.tensor(input['ids'])
+    mask = torch.tensor(inputs['mask'])
+    token_type = torch.tensor(inputs['token_type'])
     offsets = inputs['offset']
     expansion = inputs['expnsion']
     acronym = inputs['acronym']
 
+    ids = torch.unsqueeze(ids, 0)
+    mask = torch.unsqueeze(mask, 0)
+    token_type = torch.unsqueeze(token_type, 0)
+
     start_logits, end_logits = MODEL(ids, mask, token_type)
 
     start_prob = torch.softmax(start_logits, axis=-1).detach().numpy()