Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
495f9d5
1
Parent(s):
151727a
Increase duration
Browse files
app.py
CHANGED
@@ -46,7 +46,7 @@ def image_to_base64(image_path):
|
|
46 |
return f"data:image/jpeg;base64,{encoded_string}"
|
47 |
|
48 |
|
49 |
-
@spaces.GPU(duration=
|
50 |
def pixtral_inference(
|
51 |
images,
|
52 |
text,
|
@@ -84,7 +84,7 @@ def pixtral_inference(
|
|
84 |
return result
|
85 |
|
86 |
|
87 |
-
@spaces.GPU(duration=
|
88 |
def retrieve(query: str, ds, images, k):
|
89 |
if len(images) == 0:
|
90 |
raise gr.Error("No docs/images for retrieval")
|
@@ -136,7 +136,7 @@ def convert_files(files):
|
|
136 |
return images
|
137 |
|
138 |
|
139 |
-
@spaces.GPU(duration=
|
140 |
def index_gpu(images, ds):
|
141 |
model = ColPali.from_pretrained(
|
142 |
COLPALI_GEMMA_MODEL_PATH,
|
|
|
46 |
return f"data:image/jpeg;base64,{encoded_string}"
|
47 |
|
48 |
|
49 |
+
@spaces.GPU(duration=45)
|
50 |
def pixtral_inference(
|
51 |
images,
|
52 |
text,
|
|
|
84 |
return result
|
85 |
|
86 |
|
87 |
+
@spaces.GPU(duration=45)
|
88 |
def retrieve(query: str, ds, images, k):
|
89 |
if len(images) == 0:
|
90 |
raise gr.Error("No docs/images for retrieval")
|
|
|
136 |
return images
|
137 |
|
138 |
|
139 |
+
@spaces.GPU(duration=45)
|
140 |
def index_gpu(images, ds):
|
141 |
model = ColPali.from_pretrained(
|
142 |
COLPALI_GEMMA_MODEL_PATH,
|