Update app.py

app.py

@@ -344,9 +344,11 @@ def restore_on_gpu(
     torch.cuda.set_device(SUPIR_device)
 
     with torch.no_grad():
-        input_image = Image.open(input_image_path)
+        # Convert input image to NumPy array and ensure it has 3 channels
+        input_image = HWC3(np.array(Image.open(input_image_path)))
+
         input_image = upscale_image(input_image, upscale, unit_resolution=32, min_size=min_size)
-        LQ = np.array(input_image) / 255.0
+        LQ = input_image / 255.0
         LQ = np.power(LQ, gamma_correction)
         LQ *= 255.0
         LQ = LQ.round().clip(0, 255).astype(np.uint8)
@@ -367,8 +369,8 @@ def restore_on_gpu(
         results = [x_samples[i] for i in range(num_samples)]
     torch.cuda.empty_cache()
 
-    input_height, input_width, input_channel = np.array(input_image).shape
-    result_height, result_width, result_channel = np.array(results[0]).shape
+    input_height, input_width, input_channel = input_image.shape
+    result_height, result_width, result_channel = results[0].shape
 
     print('Restoration completed.')
     end = time.time()