Upload gradio_s3diff.py

src/gradio_s3diff.py

@@ -17,6 +17,7 @@ from de_net import DEResNet
 from s3diff_tile import S3Diff
 from torchvision import transforms
 from utils.wavelet_color import wavelet_color_fix, adain_color_fix
+from spaces import GPU  # 追加
 
 tensor_transforms = transforms.Compose([
                 transforms.ToTensor(),
@@ -65,6 +66,7 @@ device = "cuda"
 net_sr.to(device, dtype=weight_dtype)
 net_de.to(device, dtype=weight_dtype)
 
+@GPU(duration=60)  # GPUを利用する関数にデコレーターを追加
 @torch.no_grad()
 def process(
     input_image: Image.Image,