Update app.py

app.py

@@ -20,7 +20,7 @@ pipe = DiffusionPipeline.from_pretrained(
 #generator = torch.Generator("cuda").manual_seed(0)
 
 def infer(prompt, init_image):
-    init_image = Image.open(BytesIO(init_image)).convert("RGB")
+    init_image = Image.open(init_image).convert("RGB")
     init_image = init_image.resize((512, 512))
     
     res = pipe.train(
@@ -101,7 +101,7 @@ with gr.Blocks(css=css) as block:
         gr.HTML(title)
 
         prompt_input = gr.Textbox(label="Target text", placeholder="Describe the image with what you want to change about the subject")
-        image_init = gr.Image(source="upload", type="binary",label="Input Image")
+        image_init = gr.Image(source="upload", type="filepath",label="Input Image")
         
         submit_btn = gr.Button("Submit")