add zerogpu support

app.py

@@ -2,11 +2,13 @@ import gradio as gr
 from uform import gen_model
 from PIL import Image
 import torch
+import spaces
 
 # Load the model and processor
 model = gen_model.VLMForCausalLM.from_pretrained("unum-cloud/uform-gen")
 processor = gen_model.VLMProcessor.from_pretrained("unum-cloud/uform-gen")
 
+@spaces.GPU
 def generate_caption(image, prompt):
     # Process the image and the prompt
     inputs = processor(texts=[prompt], images=[image], return_tensors="pt")