Update app.py

app.py

@@ -8,7 +8,7 @@ import copy
 import math
 import time
 import random
-import spaces
+#import spaces
 import re
 import uuid
 
@@ -103,7 +103,7 @@ def check(input_image):
     if input_image is None:
         raise gr.Error("Please provide an image to restore.")
 
-@spaces.GPU(duration=420)
+#@spaces.GPU(duration=420)
 def stage1_process(
     input_image,
     gamma_correction,
@@ -310,43 +310,43 @@ def restore_in_Xmin(
             noisy_image, denoise_image, prompt, a_prompt, n_prompt, num_samples, min_size, downscale, upscale, edm_steps, s_stage1, s_stage2, s_cfg, randomize_seed, seed, s_churn, s_noise, color_fix_type, diff_dtype, ae_dtype, gamma_correction, linear_CFG, linear_s_stage2, spt_linear_CFG, spt_linear_s_stage2, model_select, output_format, allocation
         )
 
-@spaces.GPU(duration=59)
+#@spaces.GPU(duration=59)
 def restore_in_1min(*args, **kwargs):
     return restore_on_gpu(*args, **kwargs)
 
-@spaces.GPU(duration=119)
+#@spaces.GPU(duration=119)
 def restore_in_2min(*args, **kwargs):
     return restore_on_gpu(*args, **kwargs)
 
-@spaces.GPU(duration=179)
+#@spaces.GPU(duration=179)
 def restore_in_3min(*args, **kwargs):
     return restore_on_gpu(*args, **kwargs)
 
-@spaces.GPU(duration=239)
+#@spaces.GPU(duration=239)
 def restore_in_4min(*args, **kwargs):
     return restore_on_gpu(*args, **kwargs)
 
-@spaces.GPU(duration=299)
+#@spaces.GPU(duration=299)
 def restore_in_5min(*args, **kwargs):
     return restore_on_gpu(*args, **kwargs)
 
-@spaces.GPU(duration=359)
+#@spaces.GPU(duration=359)
 def restore_in_6min(*args, **kwargs):
     return restore_on_gpu(*args, **kwargs)
 
-@spaces.GPU(duration=419)
+#@spaces.GPU(duration=419)
 def restore_in_7min(*args, **kwargs):
     return restore_on_gpu(*args, **kwargs)
 
-@spaces.GPU(duration=479)
+#@spaces.GPU(duration=479)
 def restore_in_8min(*args, **kwargs):
     return restore_on_gpu(*args, **kwargs)
 
-@spaces.GPU(duration=539)
+#@spaces.GPU(duration=539)
 def restore_in_9min(*args, **kwargs):
     return restore_on_gpu(*args, **kwargs)
 
-@spaces.GPU(duration=599)
+#@spaces.GPU(duration=599)
 def restore_in_10min(*args, **kwargs):
     return restore_on_gpu(*args, **kwargs)