Update app.py

app.py

@@ -59,7 +59,9 @@ model = ColPali.from_pretrained(
 ).eval()
 model.load_adapter(model_name)
 processor = AutoProcessor.from_pretrained(model_name, token=token)
-device = model.device
+device = "cuda:0" if torch.cuda.is_available() else "cpu"
+    if device != model.device:
+        model.to(device)
 mock_image = Image.new("RGB", (448, 448), (255, 255, 255))
 
 with gr.Blocks() as demo: