Update inf.py

inf.py

@@ -105,6 +105,7 @@ class InferencePipeline:
             seed: int,
             n_steps: int,
             guidance_scale: float,
+            num_images_per_prompt: int = 1
     ) -> PIL.Image.Image:
         if not torch.cuda.is_available():
             raise gr.Error('CUDA is not available.')
@@ -117,5 +118,6 @@ class InferencePipeline:
             num_inference_steps=n_steps,
             guidance_scale=guidance_scale,
             generator=generator,
+            num_images_per_prompt=num_images_per_prompt,
         )  # type: ignore
-        return out.images[0]
+        return out.images