Update app.py

app.py

@@ -55,9 +55,13 @@ def index(files, ds):
         collate_fn=lambda x: process_images(processor, x),
     )
 
+    print(f"model device: {model.device})
+          
     for batch_doc in tqdm(dataloader):
         with torch.no_grad():
             batch_doc = {k: v.to(model.device) for k, v in batch_doc.items()}
+            print(f"model device: {model.device})
+            print(f"model device: {batch_doc.input_ids})
             embeddings_doc = model(**batch_doc)
         ds.extend(list(torch.unbind(embeddings_doc.to("cpu"))))
     return f"Uploaded and converted {len(images)} pages", ds, images
@@ -71,8 +75,6 @@ model = ColPali.from_pretrained(
 model.load_adapter(model_name)
 processor = AutoProcessor.from_pretrained(model_name, token = token)
 
-device = model.device
-
 mock_image = Image.new("RGB", (448, 448), (255, 255, 255))
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo: