Update app.py

app.py

@@ -22,7 +22,6 @@ import io
 from io import BytesIO
 from botocore.exceptions import NoCredentialsError, PartialCredentialsError
 
-torch.cuda.empty_cache()
 
 app = FastAPI()
 
@@ -40,7 +39,8 @@ if torch.cuda.is_available():
     device = "cuda:0"
 else:
     device = "cpu"
-    
+
+torch.cuda.synchronize()
 
 model = TSR.from_pretrained(
     "stabilityai/TripoSR",
@@ -182,8 +182,7 @@ def preprocess(input_image, do_remove_background, foreground_ratio):
 
 # @spaces.GPU
 def generate(image, mc_resolution, formats=["obj", "glb"]):
-    torch.cuda.synchronize()
-    torch.cuda.empty_cache()
+    # torch.cuda.empty_cache()
     scene_codes = model(image, device=device)
     mesh = model.extract_mesh(scene_codes, resolution=mc_resolution)[0]
     mesh = to_gradio_3d_orientation(mesh)
@@ -194,9 +193,6 @@ 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):