Update app.py

app.py

@@ -208,6 +208,7 @@ def load_model(model_name):
 # Dictionary to store loaded models
 loaded_models = {}
 
+@spaces.GPU
 def process_image(image, model_name, confidence_threshold):
     """Main processing function for Gradio interface"""
     global loaded_models