Updated app

app.py

@@ -5,7 +5,7 @@ import numpy as np
 from PIL import Image
 from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
 
-MAX_IMAGES = 4
+MAX_IMAGES = 1
 
 
 def generate_images(
@@ -40,7 +40,7 @@ def generate_images(
             prompt,
             height=288,
             width=288,
-            num_inference_steps=25,
+            num_inference_steps=10,
             guidance_scale=7.5,
             cross_attention_kwargs={"scale": 1.0},
         ).images[0]
@@ -58,7 +58,7 @@ lora_model_path = "michaelriedl/MonsterForgeFusion-sd-2-base"
 
 # Create the pipeline
 pipe = StableDiffusionPipeline.from_pretrained(
-    model_base, torch_dtype=torch.float16, use_safetensors=False, local_files_only=False
+    model_base, torch_dtype=torch.float32, use_safetensors=False, local_files_only=False
 )
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 pipe.unet.load_attn_procs(lora_model_path)
@@ -77,7 +77,7 @@ with demo:
     with gr.Column():
         with gr.Row():
             gallery = gr.Gallery(
-                columns=4,
+                columns=MAX_IMAGES,
                 object_fit="scale-down",
             )
         with gr.Row():