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 = []
|