Update app.py

app.py

@@ -41,7 +41,10 @@ def infer(init_img, prompt, negative_prompt, seed, randomize_seed, width, height
 
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
-        
+
+    init_img = Image.open(init_img).convert("RGB")
+    init_image.thumbnail((768, 768))
+    
     generator = torch.Generator().manual_seed(seed)
     
     image = pipe(
@@ -85,7 +88,7 @@ with gr.Blocks(css=css) as demo:
 
 
         with gr.Row():
-            init_img = gr.Image()            
+            init_img = gr.Image(type="pil")
 
         with gr.Row():