test-deepsearch

Sleeping

manu commited on Jun 25, 2024

Commit

d5db6a5

verified ·

1 Parent(s): 46fe397

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -85,7 +85,7 @@ def index(file, ds):
 COLORS = ["#4285f4", "#db4437", "#f4b400", "#0f9d58", "#e48ef1"]
 # Load model
-model_name = "vidore/colpali-3b-mix-448"
 token = os.environ.get("HF_TOKEN")
 model = ColPali.from_pretrained(
     "google/paligemma-3b-mix-448", torch_dtype=torch.bfloat16, device_map="cuda", token=token

 COLORS = ["#4285f4", "#db4437", "#f4b400", "#0f9d58", "#e48ef1"]
 # Load model
+model_name = "vidore/colpali"
 token = os.environ.get("HF_TOKEN")
 model = ColPali.from_pretrained(
     "google/paligemma-3b-mix-448", torch_dtype=torch.bfloat16, device_map="cuda", token=token