Update app.py

app.py

@@ -24,7 +24,8 @@ os.chdir("Stable_Diffusion_Finetuned_Minecraft_Skin_Generator")
 
 @spaces.GPU()
 def run_inference(prompt, stable_diffusion_model, num_inference_steps, guidance_scale, model_precision_type, seed, filename, verbose, see_in_3d):
-    
+
+    # inference
     if stable_diffusion_model == '2':
         sd_model = "minecraft-skins"
     else:
@@ -34,7 +35,7 @@ def run_inference(prompt, stable_diffusion_model, num_inference_steps, guidance_
     
     os.system(inference_command)
 
-    # view it in 3d
+    # view it in 3d or not
     if see_in_3d:
         os.chdir("Scripts")
         command_3d_model = f"python to_3d_model.py '{filename}'"
@@ -54,45 +55,28 @@ guidance_scale = gr.Number(minimum=0.1, value=7.5, label="Guidance Scale", info=
 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 gives better results")
 seed = gr.Number(value=42, label="Seed", info="A starting point to initiate generation, put 0 for a random one")
 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")
-verbose = gr.Checkbox(label="Verbose Output", info="Produce more detailed output while running", value=False)
 see_in_3d = gr.Checkbox(label="See in 3D", info="View the generated skin in 3D", value=False)
-
-title="<h1>Minecraft Skin Generator</h1>"
-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>Credits: [Monadical-SAS](https://github.com/Monadical-SAS/minecraft_skin_generator) (Creators of the model), [Nick088](https://linktr.ee/Nick088) (Improving usage of the model), daroche (helping me fix the 3d model texture isue), [Brottweiler](https://gist.github.com/Brottweiler/483d0856c6692ef70cf90bf1a85ce364)(script to fix the 3d model texture, [meew](https://huggingface.co/spaces/meeww/Minecraft_Skin_Generator/blob/main/models/player_model.glb) (Minecraft Player 3d model)"
-
-with gr.Blocks() as demo:
-    gr.Markdown(title)
-    gr.Markdown(description)
-    with gr.Column() as inference:
-        inputs=[
-            prompt,
-            stable_diffusion_model,
-            num_inference_steps,
-            guidance_scale,
-            model_precision_type,
-            seed,
-            filename,
-            verbose,
-            see_in_3d
-        ]
-
-        submit_btn = gr.Button("Submit")
-
-        
-    with gr.Column() as outputs:
-        outputs=[
-            gr.Image(label="Generated Minecraft Skin Image Asset"),
-            gr.Model3D(clear_color=[0.0, 0.0, 0.0, 0.0],  label="3D Model")
-        ]
-    
-
-    submit_btn.click(
-        run_inference,
-        inputs,
-        outputs,
-    )
-
-    
+verbose = gr.Checkbox(label="Verbose Output", info="Produce more detailed output while running", value=False)
 
 
-demo.launch(show_api=False, share=True)
+# Create the Gradio interface
+gr.Interface(
+    fn=run_inference,
+    inputs=[
+        prompt,
+        stable_diffusion_model,
+        num_inference_steps,
+        guidance_scale,
+        model_precision_type,
+        seed,
+        filename,
+        see_in_3d
+        verbose
+    ],
+    outputs=[
+        gr.Image(label="Generated Minecraft Skin Image Asset"),
+        gr.Model3D(clear_color=[0.0, 0.0, 0.0, 0.0],  label="3D Model")
+    ],
+    title="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>Credits: [Monadical-SAS](https://github.com/Monadical-SAS/minecraft_skin_generator) (Creators of the model), [Nick088](https://linktr.ee/Nick088) (Improving usage of the model), daroche (helping me fix the 3d model texture isue), [Brottweiler](https://gist.github.com/Brottweiler/483d0856c6692ef70cf90bf1a85ce364)(script to fix the 3d model texture, [meew](https://huggingface.co/spaces/meeww/Minecraft_Skin_Generator/blob/main/models/player_model.glb) (Minecraft Player 3d model)",
+).launch(show_api=False, share=True)