Update app.py

app.py

@@ -4,6 +4,8 @@ import open_clip
 import gradio as gr
 from mobileclip.modules.common.mobileone import reparameterize_model
 
+import spaces
+
 ################################################################################
 # Model Configuration
 ################################################################################
@@ -51,6 +53,7 @@ def load_model(model_name):
 ################################################################################
 # Inference
 ################################################################################
+@spaces.GPU(duration=120)
 def classify_image(image, candidate_labels, model_name):
     """
     Classify image using selected MobileCLIP2 model