manu commited on
Commit
16bb19e
1 Parent(s): 084583c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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))