fix types

app.py

@@ -105,7 +105,7 @@ def resize_style(img: Image.Image) -> Image.Image:
 
 
 def style_sofa(
-    Input_image: np.ndarray, Style_image: np.ndarray, Choice_of_algorithm: str
+    Input_image: Image.Image, Style_image: Image.Image, Choice_of_algorithm: str
 ) -> Image.Image:
     """
     Styles (all) the sofas in the image to the given style.
@@ -122,9 +122,8 @@ def style_sofa(
     id = randint(0, 10)
     print("Starting job ", id)
     # preprocess input images to fit requirements of the segmentation model
-    input_img, style_img = Image.fromarray(Input_image), Image.fromarray(Style_image)
-    resized_img, box = resize_sofa(fix_orient(input_img))
-    resized_style = resize_style(fix_orient(style_img))
+    resized_img, box = resize_sofa(fix_orient(Input_image))
+    resized_style = resize_style(fix_orient(Style_image))
     # resized_style.save('resized_style.jpg')
     # generate mask for image
     print("generating mask...")
@@ -145,8 +144,8 @@ def style_sofa(
 demo = gr.Interface(
     style_sofa,
     inputs=[
-        gr.inputs.Image(),
-        gr.inputs.Image(),
+        gr.inputs.Image(type="pil"),
+        gr.inputs.Image(type="pil"),
         gr.inputs.Radio(
             ["Style Transformer", "Style FAST", "Style Projection"],
             default="Style FAST",