Update app.py

app.py

@@ -7,12 +7,14 @@ from PIL import Image
 from torch.utils.data import DataLoader
 from tqdm import tqdm
 from transformers import AutoProcessor
+import spaces
 
 from colpali_engine.models.paligemma_colbert_architecture import ColPali
 from colpali_engine.trainer.retrieval_evaluator import CustomEvaluator
 from colpali_engine.utils.colpali_processing_utils import process_images, process_queries
 
 
+@spaces.GPU
 def search(query: str, ds, images):
     qs = []
     with torch.no_grad():
@@ -28,6 +30,7 @@ def search(query: str, ds, images):
     return f"The most relevant page is {best_page}", images[best_page]
 
 
+@spaces.GPU
 def index(file, ds):
     """Example script to run inference with ColPali"""
     images = []