Update app.py
Browse files
app.py
CHANGED
|
@@ -55,7 +55,7 @@ def index(files, ds):
|
|
| 55 |
collate_fn=lambda x: process_images(processor, x),
|
| 56 |
)
|
| 57 |
|
| 58 |
-
print(f"model device: {model.device})
|
| 59 |
|
| 60 |
for batch_doc in tqdm(dataloader):
|
| 61 |
with torch.no_grad():
|
|
|
|
| 55 |
collate_fn=lambda x: process_images(processor, x),
|
| 56 |
)
|
| 57 |
|
| 58 |
+
print(f"model device: {model.device}")
|
| 59 |
|
| 60 |
for batch_doc in tqdm(dataloader):
|
| 61 |
with torch.no_grad():
|