Fix dataloader input

inference.py

@@ -81,11 +81,11 @@ model = torch.load(model_path, map_location=device)
 
 model.eval()
 
-train_sampler = RandomSampler(data_sample)
-train_dataloader = DataLoader(data_sample, sampler=train_sampler, batch_size=1)
+train_sampler = RandomSampler(input_as_dataset)
+train_dataloader = DataLoader(input_as_dataset, sampler=train_sampler, batch_size=1)
 
 for batch in train_dataloader:
-	print(batch)
+	#print(batch)
 	inputs = {'input_ids': batch[0],
 					'attention_mask': batch[1],
 					'token_type_ids': batch[2],