Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -22,6 +22,7 @@ model = ColPali.from_pretrained(
|
|
22 |
"google/paligemma-3b-mix-448", torch_dtype=torch.bfloat16, device_map="cuda", token = token).eval()
|
23 |
|
24 |
model.load_adapter(model_name)
|
|
|
25 |
processor = AutoProcessor.from_pretrained(model_name, token = token)
|
26 |
|
27 |
mock_image = Image.new("RGB", (448, 448), (255, 255, 255))
|
|
|
22 |
"google/paligemma-3b-mix-448", torch_dtype=torch.bfloat16, device_map="cuda", token = token).eval()
|
23 |
|
24 |
model.load_adapter(model_name)
|
25 |
+
model = model.eval()
|
26 |
processor = AutoProcessor.from_pretrained(model_name, token = token)
|
27 |
|
28 |
mock_image = Image.new("RGB", (448, 448), (255, 255, 255))
|