Update app.py

app.py

@@ -112,7 +112,7 @@ def inference(image, face_align, background_enhance, face_upsample, upscale, cod
         draw_box = False
         detection_model = "retinaface_resnet50"
 
-        print('Inp:', image, background_enhance, face_upsample, upscale, codeformer_fidelity)
+        print('Inp:', image.shape, background_enhance, face_upsample, upscale, codeformer_fidelity)
         face_align = face_align if face_align is not None else True
         background_enhance = background_enhance if background_enhance is not None else True
         face_upsample = face_upsample if face_upsample is not None else True
@@ -121,8 +121,12 @@ def inference(image, face_align, background_enhance, face_upsample, upscale, cod
         has_aligned = not face_align
         upscale = 1 if has_aligned else upscale
 
-        img = cv2.imread(str(image), cv2.IMREAD_COLOR)
-        print('\timage size:', img.shape)
+        img = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
+        
+        if img is None:
+            raise ValueError(f"Failed to read image from {image_path}")
+    
+        # print('\timage size:', img.shape)
 
         upscale = int(upscale) # convert type to int
         if upscale > 4: # avoid memory exceeded due to too large upscale
@@ -280,7 +284,7 @@ td {
 
 demo = gr.Interface(
     inference, [
-        gr.Image(type="filepath", label="Input"),
+        gr.Image(type="numpy", label="Input"),
         gr.Checkbox(value=True, label="Pre_Face_Align"),
         gr.Checkbox(value=True, label="Background_Enhance"),
         gr.Checkbox(value=True, label="Face_Upsample"),