added image.save() using slugify and mkdir

app.py

@@ -1,7 +1,8 @@
 import torch
 from diffusers import DiffusionPipeline, AutoencoderKL
 import gradio as gr
-
+from pathlib import Path
+from slugify import slugify
 
 # Function to load the model.
 def load_model():
@@ -31,14 +32,25 @@ def generate_image(prompt, num_inference_steps):
     num_inference_steps_int = int(num_inference_steps)
     
     image = pipe(prompt, num_inference_steps=num_inference_steps_int).images[0]
-    image.save("msif.jpg")
+
+    # save the generated image
+    DIR_NAME="./images/"
+    dirpath = Path(DIR_NAME)
+    # create parent dir if doesn't exist
+    dirpath.mkdir(parents=True, exist_ok=True)
+    # create filename for image based on prompt
+    image_name = f'{slugify(prompt)}.jpg'
+    image_path = dirpath / image_name
+    image.save(image_path)
+    print(image_name)
+    
     return image
 
 iface = gr.Interface(
     fn=generate_image,
     inputs=[
         gr.Textbox(label="Enter a prompt for the image"),
-        gr.Textbox(label="Number of Inference Steps")
+        gr.Textbox(label="Number of Inference Steps min=10 or max=50")
     ],
     outputs="image",
     title="Stable Diffusion XL Text2Image Finetune Dreambooth",