Update app.py

app.py

@@ -125,12 +125,16 @@ def generate_3D(input, model_name='base40M', guidance_scale=3.0, grid_size=32):
     set_state('Converting to point cloud...')
     pc = sampler.output_to_point_clouds(samples)[0]
 
+    set_state('Saving point cloud...')
+    with open("point_cloud.ply", "wb") as f:
+        pc.write_ply(f)
+
     set_state('Converting to mesh...')
-    save_ply(pc, 'point_cloud.ply', grid_size)
+    save_ply(pc, 'mesh.ply', grid_size)
 
     set_state('')
 
-    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)
+    return pc_to_plot(pc), ply_to_obj('point_cloud.ply', '3d_model.obj'), gr.update(value=['3d_model.obj', 'mesh.ply', 'point_cloud.ply'], visible=True)
 
 def prepare_img(img):
 
@@ -188,7 +192,10 @@ def save_ply(pc, file_name, grid_size):
 with gr.Blocks() as app:
     gr.Markdown("## Point-E text-to-3D Demo")
     gr.Markdown("This is a demo for [Point-E: A System for Generating 3D Point Clouds from Complex Prompts](https://arxiv.org/abs/2212.08751) by OpenAI. Check out the [GitHub repo](https://github.com/openai/point-e) for more information.")
-
+    gr.HTML("""To skip the queue you can duplicate this space:
+            <br><a href="https://huggingface.co/spaces/anzorq/point-e_demo?duplicate=true"><img src="https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14" alt="Duplicate Space"></a>
+            <br>Don't forget to change space hardware to <b>GPU</b> after duplicating it.""")
+    
     with gr.Row():
         with gr.Column():
             with gr.Tab("Text to 3D"):