Update app.py

app.py

@@ -442,6 +442,10 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
     generator = torch.Generator(device="cuda").manual_seed(seed)
     final_image = None  # Initialize final_image
     with calculateDuration("Generating image"):
+        # Ensure width and height are integers
+        width = int(width) if isinstance(width, (float, int)) else 1024
+        height = int(height) if isinstance(height, (float, int)) else 1024
+
         # Generate image
         for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
             prompt=prompt_mash,