Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 import torch
+import spaces
 from transformers import AutoModel, AutoProcessor
+
 from gender_classification import gender_classification
 from emotion_classification import emotion_classification
 from dog_breed import dog_breed_classification
@@ -72,6 +74,7 @@ siglip2_model = AutoModel.from_pretrained(sg2_ckpt, device_map="cpu").eval()
 siglip2_processor = AutoProcessor.from_pretrained(sg2_ckpt)
 
 # Utilities for zero-shot classification.
+@spaces.GPU
 def postprocess_siglip(sg1_probs, sg2_probs, labels):
     sg1_output = {labels[i]: sg1_probs[0][i].item() for i in range(len(labels))}
     sg2_output = {labels[i]: sg2_probs[0][i].item() for i in range(len(labels))}