Update app.py

app.py

@@ -73,7 +73,13 @@ CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "0")
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
-NUM_IMAGES_PER_PROMPT = 6  # Set maximum images per prompt
+# Initialize the DiffusionPipeline (this was missing)
+pipe = DiffusionPipeline.from_pretrained(
+    "SG161222/RealVisXL_V3.0_Turbo",  # or any model of your choice
+    torch_dtype=torch.float16,
+    use_safetensors=True,
+    variant="fp16"
+).to(device)
 
 # Convert mm to pixels for a specific DPI (300)
 def mm_to_pixels(mm, dpi=300):
@@ -150,7 +156,7 @@ def generate(
         "guidance_scale": guidance_scale,
         "num_inference_steps": 25,
         "generator": generator,
-        "num_images_per_prompt": NUM_IMAGES_PER_PROMPT,
+        "num_images_per_prompt": 6,  # Max 6 images
         "output_type": "pil",
     }