Update app.py

app.py

@@ -53,16 +53,28 @@ def search(query: str, ds, images, k):
     return results
 
 
-@spaces.GPU
 def index(files, ds):
-    """Example script to run inference with ColPali"""
+    print("Converting files")
+    images = convert_files(files)
+    print(f"Files converted with {len(images)} images.")
+    return index_gpu(images, ds)
+    
+
+
+def convert_files(files):
     images = []
     for f in files:
-        images.extend(convert_from_path(f))
+        images.extend(convert_from_path(f, thread_count=4))
 
     if len(images) >= 150:
         raise gr.Error("The number of images in the dataset should be less than 150.")
+    return images
 
+
+@spaces.GPU
+def index_gpu(images, ds):
+    """Example script to run inference with ColPali"""
+    
     # run inference - docs
     dataloader = DataLoader(
         images,