add zero

app.py

@@ -11,6 +11,7 @@ import flax.linen as nn
 import jax
 import jax.numpy as jnp
 import numpy as np
+import spaces
 
 
 hf_token = os.getenv("HF_TOKEN")
@@ -21,7 +22,7 @@ model = PaliGemmaForConditionalGeneration.from_pretrained(model_id, use_auth_tok
 processor = PaliGemmaProcessor.from_pretrained(model_id)
 
 ###### Transformers Inference
-
+@spaces.GPU
 def infer(
     image: PIL.Image.Image,
     text: str,