Update app.py

app.py

@@ -3,12 +3,14 @@ from transformers import AutoProcessor, LlavaForConditionalGeneration
 from PIL import Image
 import torch
 import numpy as np
+import spaces
 
 # Load the Pixtral model and processor
 model_id = "mrcuddle/lumimaid-v0.2-8b-pixtral"
 processor = AutoProcessor.from_pretrained(model_id)
 model = LlavaForConditionalGeneration.from_pretrained(model_id, ignore_mismatched_sizes=True).to("cuda")
 
+@spaces.GPU
 def generate_text(input_text="", image=None):
     if image is None:
         return "Please upload an image."