Update app.py

app.py

@@ -3,6 +3,7 @@ from PIL import Image
 import numpy as np
 from transformers import AutoModel
 import torch
+import spaces
 
 # Load the model
 model = AutoModel.from_pretrained("ragavsachdeva/magiv2", trust_remote_code=True).cuda().eval()
@@ -12,6 +13,7 @@ def read_image(image):
     image = np.array(image)
     return image
 
+@spaces.GPU
 def process_images(chapter_pages, character_bank_images, character_bank_names):
     chapter_pages = [read_image(image) for image in chapter_pages]
     character_bank = {