HUANG-Stephanie commited on
Commit
2902a60
1 Parent(s): d005da4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,6 +1,8 @@
1
  import os
2
  import sys
3
 
 
 
4
  import gradio as gr
5
  import torch
6
  from colpali_engine.models.paligemma_colbert_architecture import ColPali
@@ -16,8 +18,6 @@ from torch.utils.data import DataLoader
16
  from tqdm import tqdm
17
  from transformers import AutoProcessor
18
 
19
- sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), './colpali-main')))
20
-
21
  # Load model
22
  model_name = "vidore/colpali"
23
  token = os.environ.get("HF_TOKEN")
 
1
  import os
2
  import sys
3
 
4
+ sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), './colpali-main')))
5
+
6
  import gradio as gr
7
  import torch
8
  from colpali_engine.models.paligemma_colbert_architecture import ColPali
 
18
  from tqdm import tqdm
19
  from transformers import AutoProcessor
20
 
 
 
21
  # Load model
22
  model_name = "vidore/colpali"
23
  token = os.environ.get("HF_TOKEN")