Reactivate Stage 1

gradio_demo.py

@@ -69,9 +69,6 @@ if torch.cuda.device_count() > 0:
 
 @spaces.GPU(duration=120)
 def stage1_process(input_image, gamma_correction):
-    return None
-
-def stage1_process2(input_image, gamma_correction):
     if torch.cuda.device_count() == 0:
         gr.Warning('Set this space to GPU config to make it work.')
         return None