fix: spaces gpu

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 
-# import spaces
+import spaces
 import torch
 from PIL import Image
 import numpy as np
@@ -12,9 +12,8 @@ print(zero.device)  # <-- 'cpu' 🤔
 lama: SimpleLama = SimpleLama(device=zero.device)
 
 
-# @spaces.GPU
+@spaces.GPU
 def lama_inpainting(image: np.ndarray, mask: np.ndarray) -> np.ndarray:
-    print(image.shape, mask.shape)
     res = lama(
         Image.fromarray(cv2.cvtColor(image, cv2.COLOR_BGR2RGB)).convert("RGB"),
         Image.fromarray(cv2.cvtColor(mask, cv2.COLOR_BGR2RGB)).convert("L"),