Update model.py

model.py

@@ -2,7 +2,6 @@ import gc
 
 import numpy as np
 import pandas as pd
-from tqdm.notebook import tqdm, trange
 
 import torch
 from torch import nn
@@ -283,7 +282,7 @@ def predict_location_preds(tokenizer, model, feature_text, pn_history, pn_histor
     preds = []
 
     with torch.no_grad():
-        for batch in tqdm(test_dl):
+        for batch in test_dl:
 
             for k, v in batch.items():
                 if k not in  ['offset_mapping', 'sequence_id']: