Spaces:

p3nguknight
/

colpali-pixtral

Running on Zero

p3nguknight commited on Sep 22

Commit

1756157

•

1 Parent(s): 27b9b37

Increase duration

Files changed (1) hide show

app.py CHANGED Viewed

@@ -84,7 +84,7 @@ def pixtral_inference(
     return result
-@spaces.GPU(duration=45)
 def retrieve(query: str, ds, images, k):
     if len(images) == 0:
         raise gr.Error("No docs/images for retrieval")
@@ -136,7 +136,7 @@ def convert_files(files):
     return images
-@spaces.GPU(duration=45)
 def index_gpu(images, ds):
     model = ColPali.from_pretrained(
         COLPALI_GEMMA_MODEL_PATH,

     return result
+@spaces.GPU(duration=60)
 def retrieve(query: str, ds, images, k):
     if len(images) == 0:
         raise gr.Error("No docs/images for retrieval")
     return images
+@spaces.GPU(duration=60)
 def index_gpu(images, ds):
     model = ColPali.from_pretrained(
         COLPALI_GEMMA_MODEL_PATH,