spaces gpu decorator

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import spaces
 import torch
 import yaml
 import os
@@ -80,7 +81,7 @@ def generate_noisy_image(image_choice, noise_sigma, operator_key):
 
     return noisy_image, data  # Return the noisy image and data for restoration
 
-
+@spaces.GPU
 def run_restoration(data, T, K):
     """Run the restoration process and return the restored image."""
     global model, ddim_scheduler, model_type