updating to zero gpu

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import open_clip
 import torch
@@ -48,6 +49,7 @@ if api_key is None:
 login(token=api_key)
 
 # Initialize the model and tokenizer
+@spaces.GPU
 def load_model(progress=gr.Progress()):
     progress(0, "Initializing model...")
     model_name = 'hf-hub:Marqo/marqo-ecommerce-embeddings-B'
@@ -71,6 +73,7 @@ def load_model(progress=gr.Progress()):
 model, preprocess_val, text_features = load_model()
 
 # Prediction function
+@spaces.GPU
 def predict(image, url):
     if url:
         response = requests.get(url)
@@ -94,6 +97,7 @@ def predict(image, url):
     return image, top_10_confidences
 
 # Clear function
+@spaces.GPU
 def clear_fields():
     return None, ""