Update app.py

app.py

@@ -112,7 +112,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
 
 
 def run(
-    image: dict,
+    image: np.ndarray,  # Modified this line
     prompt: str,
     negative_prompt: str,
     style_name: str = DEFAULT_STYLE_NAME,
@@ -123,7 +123,7 @@ def run(
     seed: int = 0,
     progress=gr.Progress(track_tqdm=True),
 ) -> PIL.Image.Image:
-    image = PIL.Image.fromarray(image["data"], mode="RGB")
+    image = PIL.Image.fromarray(image["data"], mode="RGB")  # Modified this line
     image = image.convert("RGB")
     image = TF.to_tensor(image) > 0.5
     image = TF.to_pil_image(image.to(torch.float32))
@@ -148,7 +148,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             with gr.Group():
-                image = gr.Sketchpad(type="pil", image_mode="L")
+                image = gr.inputs.Image()  # Modified this line
                 prompt = gr.Textbox(label="Prompt")
                 style = gr.Dropdown(label="Style", choices=STYLE_NAMES, value=DEFAULT_STYLE_NAME)
                 run_button = gr.Button("Run")
@@ -246,4 +246,4 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.queue(max_size=20).launch()
+    demo.queue(max_size=20).launch()