Spaces:
Sleeping
Sleeping
HUANG-Stephanie
commited on
Commit
•
2902a60
1
Parent(s):
d005da4
Update app.py
Browse files
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")
|