Update app.py

app.py

@@ -170,8 +170,9 @@ def preprocess(input_image, do_remove_background, foreground_ratio):
 
 # @spaces.GPU
 def generate(image, mc_resolution, formats=["obj", "glb"]):
-    # torch.cuda.empty_cache()
+    torch.cuda.synchronize()
     scene_codes = model(image, device=device)
+    torch.cuda.synchronize()
     mesh = model.extract_mesh(scene_codes, resolution=mc_resolution)[0]
     mesh = to_gradio_3d_orientation(mesh)
 
@@ -181,6 +182,7 @@ def generate(image, mc_resolution, formats=["obj", "glb"]):
     mesh_path_obj = tempfile.NamedTemporaryFile(suffix=f".obj", delete=False)
     mesh.apply_scale([-1, 1, 1])  # Otherwise the visualized .obj will be flipped
     mesh.export(mesh_path_obj.name)
+    torch.cuda.synchronize()
     return mesh_path_obj.name, mesh_path_glb.name
 
 def upload_file_to_s3(file_path, bucket_name, object_name=None):