Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -254,7 +254,6 @@ 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 |
-
print(tags)
|
| 258 |
with torch.no_grad():
|
| 259 |
for batch, filenames in dataloader:
|
| 260 |
batch = batch.to(device)
|
|
@@ -273,7 +272,7 @@ def process_images(images, threshold):
|
|
| 273 |
|
| 274 |
tags = ", ".join([t[0] for t in temp])
|
| 275 |
all_results.append((filenames[i], tags, tag_score))
|
| 276 |
-
|
| 277 |
return all_results
|
| 278 |
|
| 279 |
def is_valid_image(file_path):
|
|
|
|
| 254 |
dataloader = DataLoader(dataset, batch_size=64, num_workers=0, pin_memory=True, drop_last=False)
|
| 255 |
|
| 256 |
all_results = []
|
|
|
|
| 257 |
with torch.no_grad():
|
| 258 |
for batch, filenames in dataloader:
|
| 259 |
batch = batch.to(device)
|
|
|
|
| 272 |
|
| 273 |
tags = ", ".join([t[0] for t in temp])
|
| 274 |
all_results.append((filenames[i], tags, tag_score))
|
| 275 |
+
print(all_results)
|
| 276 |
return all_results
|
| 277 |
|
| 278 |
def is_valid_image(file_path):
|