Update app.py

app.py

@@ -2,6 +2,7 @@ import subprocess
 import os
 import gradio as gr
 import torch
+import numpy as np
 from PIL import Image, ImageEnhance
 import spaces
 
@@ -49,6 +50,7 @@ num_inference_steps = gr.Slider(label="Number of Inference Steps", info="The num
 guidance_scale = gr.Slider(label="Guidance Scale", info="Controls how much the image generation process follows the text prompt. Higher values make the image stick more closely to the input text.", minimum=0.0, maximum=10.0, value=7.5, step=0.1)
 model_precision_type = gr.Dropdown(["fp16", "fp32"], value="fp16", label="Model Precision Type", info="The precision type to load the model, like fp16 which is faster, or fp32 which is more precise but more resource consuming")
 seed = gr.Slider(value=42, minimum=0, maximum=MAX_SEED, step=1, label="Seed", info="A starting point to initiate the generation process, put 0 for a random one")
+num_images_per_prompt = gr.Slider(label="Number of Images to generate with the settings",minimum=1, maximum=4, step=1, value=1)
 filename = gr.Textbox(label="Output Image Name", info="The name of the file of the output image skin, keep the.png", value="output-skin.png")
 model_3d = gr.Checkbox(label="See as 3D Model too", info="View the generated skin as a 3D Model too", value=True)
 verbose = gr.Checkbox(label="Verbose Output", info="Produce more detailed output while running", value=False)
@@ -64,12 +66,13 @@ gr.Interface(
         guidance_scale,
         model_precision_type,
         seed,
+        num_images_per_prompt,
         filename,
         model_3d,
         verbose
     ],
     outputs=[
-        gr.Image(label="Generated Minecraft Skin Image Asset", elem_classes="pixelated checkered"),
+        gr.Gallery(label="Generated Minecraft Skin Image Assets", elem_id="gallery", elem_classes="pixelated checkered"),
         gr.Model3D(clear_color=[0.0, 0.0, 0.0, 0.0],  label="3D Model View of the Skin")
     ],
     title="Minecraft Skin Generator",