Update gradio_demo/app.py

gradio_demo/app.py

@@ -5,6 +5,7 @@ sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')
 import torch
 import numpy as np
 import gradio as gr
+import spaces
 from PIL import Image
 
 from diffusers import DDPMScheduler
@@ -113,7 +114,7 @@ def dynamic_guidance_slider(sampling_steps):
 def show_final_preview(preview_row):
     return preview_row[-1][0]
 
-# @spaces.GPU #[uncomment to use ZeroGPU]
+@spaces.GPU #[uncomment to use ZeroGPU]
 @torch.no_grad()
 def instantir_restore(
     lq, prompt="", steps=30, cfg_scale=7.0, guidance_end=1.0,