Update app.py

app.py

@@ -22,7 +22,9 @@ def generate(
     guidance_scale,
     num_images_per_prompt,
     model_precision_type,
-    seed
+    seed,
+    output_image_name,
+    verbose
 ):
     
     if verbose:
@@ -38,8 +40,10 @@ def generate(
     command = f"Python_Scripts/{sd_model}.py '{prompt}' {num_inference_steps} {guidance_scale} {num_images_per_prompt} {model_precision_type} {output_image_name} {verbose_opt}"
     
     subprocess.run(["python", "$command"])
+
+    name_output = output_image_name.value
     
-    return os.path.join(f"output_minecraft_skins/{output_image_name}")
+    return os.path.join(f"output_minecraft_skins/{name_output}")
 
 
 prompt = gr.Textbox(label="Prompt", interactive=True)
@@ -77,7 +81,7 @@ gr.Interface(
     inputs=[prompt, stable_diffusion_model, num_inference_steps, guidance_scale, num_images_per_prompt, model_precision_type, output_image_name, verbose, seed],
     outputs=gr.Image(label="Generated Minecraft Skin"),
     title="Stable Diffusion Finetuned Minecraft Skin Generator",
-    description="Make your prompts more detailed!<br>Model used: https://huggingface.co/roborovski/superprompt-v1<br>Hugging Face Space made by [Nick088](https://linktr.ee/Nick088)",
+    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)",
     examples=examples,
     concurrency_limit=20,
 ).launch(show_api=False)