Update app.py

app.py

@@ -30,7 +30,7 @@ inputs = [
 
 output = gr.Gallery(
             label="Generated images", show_label=False, elem_id="gallery"
-        ).style(grid=[int(math.ceil(steps/2))], height="auto")
+        ).style(grid=[2], height="auto")
 
 examples = [
   ["kakaobrain/karlo-v1-alpha", orig_start_prompt, orig_end_prompt, 6],