Update app.py

app.py

@@ -17,6 +17,7 @@ os.chdir("Stable_Diffusion_Finetuned_Minecraft_Skin_Generator")
 
 def generate(
     prompt,
+    stable_diffusion_model,
     num_inference_steps,
     guidance_scale,
     num_images_per_prompt,
@@ -30,13 +31,19 @@ def generate(
     else:
         verbose_opt = ''
 
-    subprocess.run(["python", "Python_Script/minecraft-skins-sdxl.py", prompt, num_inference_steps, guidance_scale, num_images_per_prompt, model_precision_type, seed, output_image_name, verbose_opt])
+    if stable_diffusion_model == 'xl':
+        sd_model = "minecraft-skins-sdxl"
+    else:
+        sd_model = "minecraft-skins"
+
+    subprocess.run(["python", f"Python_Script/{sd_model}.py", prompt, num_inference_steps, guidance_scale, num_images_per_prompt, model_precision_type, seed, output_image_name, verbose_opt])
 
     return os.path.join(f"output_minecraft_skins/{output_image_name}")
 
 
 prompt = gr.Textbox(label="Prompt", interactive=True)
 
+stable_diffusion_model = gr.Dropdown(["2", "xl"], interactive=True, label="Stable Diffusion Model", value="xl", info="Choose which Stable Diffusion Model to use, xl understands prompts better")
 
 num_inference_steps = gr.Number(value=50, minimum=1, interactive=True, label="Inference Steps", info="The number of denoising steps of the image. More denoising steps usually lead to a higher quality image at the cost of slower inference")
 
@@ -55,6 +62,7 @@ verbose = gr.Checkbox(label="Verbose Output", interactive=True, value=False, inf
 examples = [
     [
         "A man in a purple suit wearing a tophat.",
+        "xl",
         25,
         7.5,
         1,
@@ -67,7 +75,7 @@ examples = [
 
 gr.Interface(
     fn=generate,
-    inputs=[prompt, num_inference_steps, guidance_scale, num_images_per_prompt, model_precision_type, output_image_name, seed, verbose],
+    inputs=[prompt, stable_diffusion_model, num_inference_steps, guidance_scale, num_images_per_prompt, model_precision_type, output_image_name, seed, verbose],
     outputs=gr.Image(label="Generated Minecraft Skin"),
     title="Stable Diffusion Finetuned Minecraft Skin Generator",
     description="Make AI generated Minecraft Skins by a Finetuned Stable Diffusion Version!<br>Model used: https://github.com/Nick088Official/Stable_Diffusion_Finetuned_Minecraft_Skin_Generator<br>Hugging Face Space made by [Nick088](https://linktr.ee/Nick088)",