Comment to test

gradio_demo.py

@@ -73,19 +73,20 @@ def stage1_process(input_image, gamma_correction):
         gr.Warning('Set this space to GPU config to make it work.')
         return None
     torch.cuda.set_device(SUPIR_device)
-    LQ = HWC3(input_image)
-    LQ = fix_resize(LQ, 512)
-    # stage1
-    LQ = np.array(LQ) / 255 * 2 - 1
-    LQ = torch.tensor(LQ, dtype=torch.float32).permute(2, 0, 1).unsqueeze(0).to(SUPIR_device)[:, :3, :, :]
-    LQ = model.batchify_denoise(LQ, is_stage1=True)
-    LQ = (LQ[0].permute(1, 2, 0) * 127.5 + 127.5).cpu().numpy().round().clip(0, 255).astype(np.uint8)
-    # gamma correction
-    LQ = LQ / 255.0
-    LQ = np.power(LQ, gamma_correction)
-    LQ *= 255.0
-    LQ = LQ.round().clip(0, 255).astype(np.uint8)
-    return LQ
+    #LQ = HWC3(input_image)
+    #LQ = fix_resize(LQ, 512)
+    ## stage1
+    #LQ = np.array(LQ) / 255 * 2 - 1
+    #LQ = torch.tensor(LQ, dtype=torch.float32).permute(2, 0, 1).unsqueeze(0).to(SUPIR_device)[:, :3, :, :]
+    #LQ = model.batchify_denoise(LQ, is_stage1=True)
+    #LQ = (LQ[0].permute(1, 2, 0) * 127.5 + 127.5).cpu().numpy().round().clip(0, 255).astype(np.uint8)
+    ## gamma correction
+    #LQ = LQ / 255.0
+    #LQ = np.power(LQ, gamma_correction)
+    #LQ *= 255.0
+    #LQ = LQ.round().clip(0, 255).astype(np.uint8)
+    #return LQ
+    return None
 
 @spaces.GPU(duration=120)
 def llave_process(input_image, temperature, top_p, qs=None):