update

app.py

@@ -3,6 +3,7 @@ from glob import glob
 import yaml
 import os
 
+import spaces
 import gradio as gr
 import torch
 import torchvision
@@ -40,6 +41,7 @@ def prepare_model():
     return vae.to('cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu')
 
 
+@spaces.GPU
 @torch.no_grad()
 def add_censorship(input_image, mode, pixelation_block_size, blur_kernel_size, soft_edges, soft_edge_kernel_size):
     background, layers, _ = input_image.values()
@@ -93,6 +95,7 @@ def add_censorship(input_image, mode, pixelation_block_size, blur_kernel_size, s
     gr.Info("Try to donwload/copy the censored image to the `Remove censorsip' tab")
     return images[0].permute(1, 2, 0).cpu().numpy()
 
+@spaces.GPU
 @torch.no_grad()
 def remove_censorship(input_image, x1, y1, x2, y2):
     background, layers, _ = input_image.values()