Update app.py

app.py

@@ -25,19 +25,20 @@ img_pipe.to(device)
 source_img = gr.Image(source="upload", type="filepath", label="init_img | 512*512 px")
 gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
 
-def resize(value,img):
+def resize(w_val,l_val,img):
   #baseheight = value
   img = Image.open(img)
   #hpercent = (baseheight/float(img.size[1]))
   #wsize = int((float(img.size[0])*float(hpercent)))
   #img = img.resize((wsize,baseheight), Image.Resampling.LANCZOS)
-  img = img.resize((value,value), Image.Resampling.LANCZOS)
+  img = img.resize((w_val,l_val), Image.Resampling.LANCZOS)
   return img
 
+#init_image = init_image.resize((768, 512))
 
 def infer(prompt, source_img): 
          
-    source_image = resize(512, source_img)
+    source_image = resize(768, 512, source_img)
     source_image.save('source.png')
     images_list = img_pipe([prompt] * 2, init_image=source_image, strength=0.75)
     images = []