Update app.py

app.py

@@ -62,6 +62,18 @@ def img2img( path ,design,x_prompt,alt_prompt,strength,guidance_scale,steps):
   width1, height1 = img.size
   #img = img.resize((512, 704), resample=Image.Resampling.BILINEAR)
 
+###########
+    # Read the size of the image
+  width, height = img.size
+    
+    # Calculate the new size of the image, making sure that the width and height are multiples of 64
+  new_width = ((width + 63) // 64) * 64
+  new_height = ((height + 63) // 64) * 64
+    
+    # Resize the image
+  img = img.resize((new_width, new_height), resample=Image.Resampling.BILINEAR)
+
+#######
   try:
         caption, keywords = generate_caption_keywords(design)
         prompt = keywords
@@ -95,7 +107,9 @@ def img2img( path ,design,x_prompt,alt_prompt,strength,guidance_scale,steps):
           if artifact.type == generation.ARTIFACT_IMAGE:
               img2 = Image.open(io.BytesIO(artifact.binary))
               #img2 = img2.resize((width1, height1), resample=Image.Resampling.BILINEAR)
- 
+ ###
+              img2 = img2.resize((new_width, new_height), resample=Image.Resampling.BILINEAR)
+ ###             
               im1 = img2.save("new_image.jpg")
  
               print(type(img2))