Update app.py

app.py

@@ -10,10 +10,12 @@ from tqdm import tqdm
 
 from colpali_engine.models import ColQwen2, ColQwen2Processor
 
+device = "cuda:0" if torch.cuda.is_available() else "cpu"
+
 model = ColQwen2.from_pretrained(
         "manu/colqwen2-v1.0-alpha",
         torch_dtype=torch.bfloat16,
-        device_map="cuda:0",  # or "mps" if on Apple Silicon
+        device_map=device,  # or "mps" if on Apple Silicon
         # attn_implementation="flash_attention_2", # should work on A100
     ).eval()
 processor = ColQwen2Processor.from_pretrained("manu/colqwen2-v1.0-alpha")