Update app.py

app.py

@@ -8,6 +8,7 @@ from omegaconf import OmegaConf
 import subprocess
 from tqdm import tqdm
 import requests
+import spaces
 
 def download_file(url, filename):
     response = requests.get(url, stream=True)
@@ -57,6 +58,7 @@ load_state_dict(model, ckpt, strict=True)
 model.freeze()
 model.to("cuda")
 
+@spaces.GPU
 @torch.no_grad()
 def process(image, steps, t_max, t_min, color_fix_type):
     image = Image.open(image).convert("RGB")