Update app.py

app.py

@@ -9,7 +9,7 @@ def generate_image(prompt, negative_prompt="Low quality", width=512, height=512)
     # Run pipeline in inference (sample random noise and denoise)
     images = ldm([prompt], num_inference_steps=50, eta=0.3, guidance_scale=6, negative_prompts=[negative_prompt]).images
     # Resize image to desired width and height
-    resized_images = [image.resize((width, height)) for image in images]
+    resized_images = [image.resize((int(width), int(height))) for image in images]
     # Save images
     for idx, image in enumerate(resized_images):
         image.save(f"squirrel-{idx}.png")