Update app.py

app.py

@@ -35,10 +35,10 @@ def resize(width,img):
   return img
 
 def infer(prompt, init_image): 
-    #init_image = resize(512,init_image)
-    #init_image = init_image.save("init_image.png")
+    init_image = resize(512,init_image)
+    init_image.save("init_image.png")
     #image = pipe(prompt, init_image=init_image)["sample"][0]
-    images_list = pipe([prompt] * 2, init_image=init_image, strength=0.75)
+    images_list = pipe([prompt] * 2, init_image="init_image.png", strength=0.75)
     images = []
     safe_image = Image.open(r"unsafe.png")
     for i, image in enumerate(images_list["sample"]):