Update main.py

main.py

@@ -15,7 +15,7 @@ import matplotlib as mpl
 
 img_names = []
 with open('list_eval_partition.txt', 'r') as f:
-    for line in tqdm(f):
+    for line in tqdm.tqdm(f):
         img_name, dtype = line.rstrip().split(' ')
         img_names.append(img_name)
 
@@ -42,7 +42,7 @@ print('Make index .............')
 index = faiss.IndexFlatL2(512)
 
 hf = h5py.File('face_vecs_full.h5', 'r')
-for key in tqdm(hf.keys()):
+for key in tqdm.tqdm(hf.keys()):
     vec = np.array(hf.get(key))
     index.add(vec)