Update app.py

app.py

@@ -181,8 +181,11 @@ def generate(image, mc_resolution, formats=["obj", "glb"]):
     
     return mesh_path_obj.name, mesh_path_glb.name
 
-def run_example(image, seed, do_remove_background, foreground_ratio, mc_resolution, text_prompt=None):
-    image_pil = generate_image_from_text(encoded_image=image, seed=seed, pos_prompt=text_prompt)
+def run_example(image, seed, use_image, do_remove_background, foreground_ratio, mc_resolution, text_prompt=None):
+    if use_image:
+        image_pil = generate_image_from_text(encoded_image=image, seed=seed, pos_prompt=text_prompt)
+    else:
+        image_pil = image
     preprocessed = preprocess(image_pil, do_remove_background, foreground_ratio)
     mesh_name_obj, mesh_name_glb = generate(preprocessed, 256, ["obj", "glb"])
     return preprocessed, mesh_name_obj, mesh_name_glb
@@ -206,6 +209,9 @@ with gr.Blocks() as demo:
                     placeholder="Enter Positive Prompt"
                 )
                 seed = gr.Textbox(label="Random Seed", value=0)
+                use_image = gr.Checkbox(
+                        label="Enhance Image", value=True
+                    )
                 processed_image = gr.Image(label="Processed Image", interactive=False, visible=False)
             with gr.Row():
                 with gr.Group():
@@ -244,7 +250,7 @@ with gr.Blocks() as demo:
                 
     submit.click(fn=check_input_image, inputs=[input_image]).success(
         fn=run_example,
-        inputs=[input_image, seed, do_remove_background, foreground_ratio, mc_resolution, text_prompt],
+        inputs=[input_image, seed, use_image, do_remove_background, foreground_ratio, mc_resolution, text_prompt],
         outputs=[processed_image, output_model_obj, output_model_glb],
         # outputs=[output_model_obj, output_model_glb],
     )