Update app.py

+ space for zeroGPU

app.py

@@ -15,9 +15,13 @@ from PIL import Image
 from functools import partial
 from main import run_main
 import time
+import spaces
+
 LENGTH=512 #length of the square area displaying/editing images
 TRANSPARENCY = 150 # transparency of the mask in display
 
+
+@spaces.GPU
 def add_mask(mask_np_list_updated, mask_label_list):
     mask_new = np.zeros_like(mask_np_list_updated[0])
     mask_np_list_updated.append(mask_new)