wip

app.py

@@ -17,6 +17,7 @@ from point_e.util.ply_util import write_ply
 from diffusers import StableDiffusionPipeline
 
 import trimesh
+import uuid
 
 
 state = ""
@@ -116,11 +117,14 @@ def generate_3D(input, model_name='base1B', guidance_scale=3.0, grid_size=128):
     pc = sampler.output_to_point_clouds(samples)[0]
 
     set_state('Converting to mesh...')
-    save_ply(pc, '/tmp/mesh.ply')
+
+    uniqid = uuid.uuid4()
+    file_path = f'/tmp/{uniqid}.ply'
+    save_ply(pc, file_path)
 
     set_state('')
 
-    return '/tmp/mesh.ply', create_gif(pc), gr.update(value=['/tmp/mesh.ply'], visible=True)
+    return file_path, create_gif(pc), gr.update(value=['/tmp/mesh.ply'], visible=True)
 
 def prepare_img(img):
 
@@ -146,7 +150,7 @@ def ply_to_glb(ply_file, glb_file):
 
 def save_ply(pc, file_name):
     # Produce a mesh (with vertex colors)
-    with open('mesh.ply', 'wb') as f:
+    with open(file_name, 'wb') as f:
         pc.write_ply(f)