Update app.py

app.py

@@ -53,13 +53,13 @@ num_inference_steps = gr.Number(label="Number of Inference Steps", precision=0,
 guidance_scale = gr.Number(minimum=0.1, value=7.5, label="Guidance Scale", 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")
 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")
+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)
+see_in_3d = gr.Checkbox(label="See in 3D", info="View the generated skin in 3D", value=True)
 
 
 # Create the Gradio interface
-gr.Interface(
+interface = gr.Interface(
     fn=run_inference,
     inputs=[
         prompt,
@@ -68,14 +68,24 @@ gr.Interface(
         guidance_scale,
         model_precision_type,
         seed,
-        filename,
+       filename,
         verbose,
         see_in_3d
     ],
     outputs=[
-        gr.Image(label="Generated Minecraft Skin Image Asset"),
-        gr.Model3D(clear_color=[0.0, 0.0, 0.0, 0.0],  label="3D Model")
+        gr.Image(label="Generated Minecraft Skin Image Asset")
     ],
     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)
+    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)"
+)
+
+# Add a custom output component that displays the 3D model if the "See in 3D" input is True
+def custom_output(image_path, glb_path):
+    if glb_path is None:
+        return image_path
+    else:
+        return [image_path, gr.Model3D(clear_color=[0.0, 0.0, 0.0, 0.0], label="3D Model", path=glb_path)]
+
+interface.add_output(custom_output, inputs=[interface.output[0], interface.input[8]])
+
+interface.launch(show_api=False, share=True)