add support to ZeroGPU

app.py

@@ -3,6 +3,7 @@ import PIL
 import cv2
 import time
 import torch
+import spaces
 import numpy as np
 import gradio as gr
 from PIL import Image
@@ -131,7 +132,7 @@ MODEL = load_model_from_config(CONFIG, CKPT_PATH, DEVICE)
 MODEL.to(device=DEVICE)
 
 
-# @spaces.GPU(duration=60)
+@spaces.GPU(duration=60)
 def tficon(img_with_mask, ref_img, seg, prompt, dpm_order, dpm_steps, tau_a, tau_b, domain, seed, scale):
     init_img = img_with_mask["image"]
     n_samples = 1