Update app.py

app.py

@@ -93,7 +93,7 @@ def get_sampler(model_name, txt2obj, guidance_scale):
             model_kwargs_key_filter=('texts', '') if txt2obj else ("*",)
         )
 
-def generate(model_name, input, guidance_scale, grid_size):
+def generate(input, model_name='base40M', guidance_scale=3.0, grid_size=32):
 
     set_state('Entered generate function...')
 
@@ -114,11 +114,11 @@ def generate(model_name, input, guidance_scale, grid_size):
     pc = sampler.output_to_point_clouds(samples)[0]
 
     set_state('Converting to mesh...')
-    save_ply(pc, 'output.ply', grid_size)
+    save_ply(pc, 'point_cloud.ply', grid_size)
 
     set_state('')
 
-    return pc_to_plot(pc), ply_to_obj('output.ply', 'output.obj'), gr.update(value='output.obj', visible=True)
+    return pc_to_plot(pc), ply_to_obj('point_cloud.ply', '3d_model.obj'), gr.update(value=['3d_model.obj', 'point_cloud.ply'], visible=True)
 
 def prepare_img(img):
 
@@ -184,7 +184,9 @@ with gr.Blocks() as app:
                 btn_generate_txt2obj = gr.Button(value="Generate")
             with gr.Tab("Image to 3D"):
                 img = gr.Image(label="Image")
+                gr.Markdown("Best results with images of 3D objects with no shadows on a white background.")
                 btn_generate_img2obj = gr.Button(value="Generate")
+
             with gr.Accordion("Advanced settings", open=False):
                 dropdown_models = gr.Dropdown(label="Model", value="base40M", choices=["base40M"]) #, "base300M", "base1B"])
                 guidance_scale = gr.Slider(label="Guidance scale", value=3.0, minimum=3.0, maximum=10.0, step=1.0)
@@ -202,9 +204,9 @@ with gr.Blocks() as app:
         # inputs = [dropdown_models, prompt, img, guidance_scale, grid_size]
         outputs = [plot, model_3d, file_out]
 
-        prompt.submit(generate, inputs=[dropdown_models, prompt, guidance_scale, grid_size], outputs=outputs)
-        btn_generate_txt2obj.click(generate, inputs=[dropdown_models, prompt, guidance_scale, grid_size], outputs=outputs)
-        btn_generate_img2obj.click(generate, inputs=[dropdown_models, img, guidance_scale, grid_size], outputs=outputs)
+        prompt.submit(generate, inputs=[prompt, dropdown_models, guidance_scale, grid_size], outputs=outputs)
+        btn_generate_txt2obj.click(generate, inputs=[prompt, dropdown_models, guidance_scale, grid_size], outputs=outputs)
+        btn_generate_img2obj.click(generate, inputs=[img, dropdown_models, guidance_scale, grid_size], outputs=outputs)
         # btn_pc_to_obj.click(ply_to_obj, inputs=plot, outputs=[model_3d, file_out])
 
     gr.HTML("""
@@ -219,6 +221,31 @@ with gr.Blocks() as app:
     </div>
     """)
 
+    gr.Examples(
+        examples=[
+            ["a cactus in a pot"],
+            ["a round table with floral tablecloth"],
+            ["a red kettle"],
+            ["a vase with flowers"],
+            ["a sports car"],
+            ["a man"],
+        ],
+        inputs=[prompt],
+        fn=generate,
+        cache_examples=True
+    )
+
+    gr.Examples(
+        examples=[
+            ["corgi.png"],
+            ["cube_stack.jpg"],
+            ["chair.png"],
+        ],
+        inputs=[img],
+        fn=generate,
+        cache_examples=True
+    )
+
     # app.load(get_state, inputs=[], outputs=state_info, every=0.5, show_progress=False)