gpu version match

app.py

@@ -56,7 +56,7 @@ class DRACODenoiser(object):
             v2.ToDtype(torch.float32, scale=True)
         ])
 
-    @torch.inference_mode()
+    @spaces.GPU
     def inference(self, image: Image.Image) -> None:
         W, H = image.size
 
@@ -69,6 +69,7 @@ class DRACODenoiser(object):
         return denoised
 
 # Model Initialization
+@spaces.GPU
 cfg = CfgNode.load_yaml_with_base(Path("draco.yaml"))
 CfgNode.merge_with_dotlist(cfg, [])
 ckpt_path = Path("denoise.ckpt")
@@ -115,6 +116,7 @@ def display_crop(data, x_offset, y_offset, auto_contrast) -> Image:
     
     return input_image
 
+@spaces.GPU
 def process_and_denoise(data, x_offset, y_offset, auto_contrast) -> Image:
 
     crop = data[y_offset:y_offset + 1024, x_offset:x_offset + 1024]