Spaces:

HUANG-Stephanie
/

cvquest-colpali

Sleeping

HUANG-Stephanie commited on Jul 31

Commit

8450327

•

1 Parent(s): b6b56a0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -60,8 +60,8 @@ model = ColPali.from_pretrained(
 model.load_adapter(model_name)
 processor = AutoProcessor.from_pretrained(model_name, token=token)
 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:

 model.load_adapter(model_name)
 processor = AutoProcessor.from_pretrained(model_name, token=token)
 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: