Update app.py

app.py

@@ -54,6 +54,9 @@ def index(files, ds):
         shuffle=False,
         collate_fn=lambda x: process_images(processor, x),
     )
+
+    print(f"model device: {model.device}")
+    print(f"device: {device}")
     for batch_doc in tqdm(dataloader):
         with torch.no_grad():
             batch_doc = {k: v.to(device) for k, v in batch_doc.items()}