Wendy
commited on
Upload cogagent_infer_batch.py with huggingface_hub
Browse files- cogagent_infer_batch.py +1 -1
cogagent_infer_batch.py
CHANGED
@@ -141,7 +141,7 @@ def main():
|
|
141 |
data_loader = DataLoader(dataset, batch_size=16, shuffle=False)
|
142 |
|
143 |
res = []
|
144 |
-
for x in tqdm(
|
145 |
#x = data[i]
|
146 |
img_path,task = x
|
147 |
image = []
|
|
|
141 |
data_loader = DataLoader(dataset, batch_size=16, shuffle=False)
|
142 |
|
143 |
res = []
|
144 |
+
for x in tqdm(data_loader,desc="Processing items"):
|
145 |
#x = data[i]
|
146 |
img_path,task = x
|
147 |
image = []
|