cdnuts commited on
Commit
599320f
·
verified ·
1 Parent(s): cb636fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)