Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -254,7 +254,7 @@ def process_images(images, threshold):
|
|
254 |
dataloader = DataLoader(dataset, batch_size=64, num_workers=0, pin_memory=True, drop_last=False)
|
255 |
|
256 |
all_results = []
|
257 |
-
|
258 |
with torch.no_grad():
|
259 |
for batch, filenames in dataloader:
|
260 |
batch = batch.to(device)
|
|
|
254 |
dataloader = DataLoader(dataset, batch_size=64, num_workers=0, pin_memory=True, drop_last=False)
|
255 |
|
256 |
all_results = []
|
257 |
+
print(tags)
|
258 |
with torch.no_grad():
|
259 |
for batch, filenames in dataloader:
|
260 |
batch = batch.to(device)
|