Update app.py

app.py

@@ -11,14 +11,16 @@ device="cpu"
 pipe = StableDiffusionImg2ImgPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=st.secrets['USER_TOKEN'])
 pipe.to(device)
 
-def resize(value,img):
+def resize(w_val,l_val,img):
   img = Image.open(img)
-  img = img.resize((value,value), Image.Resampling.LANCZOS)
+  img = img.resize((w_val,l_val), Image.Resampling.LANCZOS)
+  #img = img.resize((value,value), Image.Resampling.LANCZOS)
   return img
 
+
 def infer(source_img, prompt, guide, steps, seed, Strength): 
     generator = torch.Generator('cpu').manual_seed(seed)     
-    source_image = resize(512, source_img)
+    source_image = resize(768, 512, source_img)
     source_image.save('source.png')
     image_list = pipe([prompt], init_image=source_image, strength=Strength, guidance_scale=guide, num_inference_steps=steps)
     images = []