p3nguknight commited on
Commit
c5aa334
·
1 Parent(s): efa909f

Try path without ~

Browse files
Files changed (1) hide show
  1. app.py +9 -4
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
- f"~/.cache/huggingface/hub/models--{PIXTAL_MODEL_ID}/{PIXTRAL_MODEL_SNAPSHOT}"
 
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 = f"~/.cache/huggingface/hub/models--{COLPALI_GEMMA_MODEL_ID}/{COLPALI_GEMMA_MODEL_SNAPSHOT}"
 
 
 
31
  COLPALI_MODEL_ID = "vidore--colpali-v1.2"
32
  COLPALI_MODEL_SNAPSHOT = "2d54d5d3684a4f5ceeefbef95df0c94159fd6a45"
33
  COLPALI_MODEL_PATH = (
34
- f"~/.cache/huggingface/hub/models--{COLPALI_MODEL_ID}/{COLPALI_MODEL_SNAPSHOT}"
 
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