Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
c5aa334
1
Parent(s):
efa909f
Try path without ~
Browse files
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import base64
|
2 |
from typing import cast
|
3 |
-
|
4 |
import gradio as gr
|
5 |
import spaces
|
6 |
import torch
|
@@ -21,17 +21,22 @@ from tqdm import tqdm
|
|
21 |
PIXTAL_MODEL_ID = "mistral-community--pixtral-12b-240910"
|
22 |
PIXTRAL_MODEL_SNAPSHOT = "95758896fcf4691ec9674f29ec90d1441d9d26d2"
|
23 |
PIXTRAL_MODEL_PATH = (
|
24 |
-
|
|
|
25 |
)
|
26 |
|
27 |
|
28 |
COLPALI_GEMMA_MODEL_ID = "vidore--colpaligemma-3b-pt-448-base"
|
29 |
COLPALI_GEMMA_MODEL_SNAPSHOT = "12c59eb7e23bc4c26876f7be7c17760d5d3a1ffa"
|
30 |
-
COLPALI_GEMMA_MODEL_PATH =
|
|
|
|
|
|
|
31 |
COLPALI_MODEL_ID = "vidore--colpali-v1.2"
|
32 |
COLPALI_MODEL_SNAPSHOT = "2d54d5d3684a4f5ceeefbef95df0c94159fd6a45"
|
33 |
COLPALI_MODEL_PATH = (
|
34 |
-
|
|
|
35 |
)
|
36 |
|
37 |
|
|
|
1 |
import base64
|
2 |
from typing import cast
|
3 |
+
import pathlib
|
4 |
import gradio as gr
|
5 |
import spaces
|
6 |
import torch
|
|
|
21 |
PIXTAL_MODEL_ID = "mistral-community--pixtral-12b-240910"
|
22 |
PIXTRAL_MODEL_SNAPSHOT = "95758896fcf4691ec9674f29ec90d1441d9d26d2"
|
23 |
PIXTRAL_MODEL_PATH = (
|
24 |
+
pathlib.Path().home()
|
25 |
+
/ f".cache/huggingface/hub/models--{PIXTAL_MODEL_ID}/{PIXTRAL_MODEL_SNAPSHOT}"
|
26 |
)
|
27 |
|
28 |
|
29 |
COLPALI_GEMMA_MODEL_ID = "vidore--colpaligemma-3b-pt-448-base"
|
30 |
COLPALI_GEMMA_MODEL_SNAPSHOT = "12c59eb7e23bc4c26876f7be7c17760d5d3a1ffa"
|
31 |
+
COLPALI_GEMMA_MODEL_PATH = (
|
32 |
+
pathlib.Path().home()
|
33 |
+
/ f".cache/huggingface/hub/models--{COLPALI_GEMMA_MODEL_ID}/{COLPALI_GEMMA_MODEL_SNAPSHOT}"
|
34 |
+
)
|
35 |
COLPALI_MODEL_ID = "vidore--colpali-v1.2"
|
36 |
COLPALI_MODEL_SNAPSHOT = "2d54d5d3684a4f5ceeefbef95df0c94159fd6a45"
|
37 |
COLPALI_MODEL_PATH = (
|
38 |
+
pathlib.Path().home()
|
39 |
+
/ f".cache/huggingface/hub/models--{COLPALI_MODEL_ID}/{COLPALI_MODEL_SNAPSHOT}"
|
40 |
)
|
41 |
|
42 |
|