Update app.py

app.py

@@ -332,96 +332,18 @@ def move_to_device(model, device):
         
 @spaces.GPU
 def extract_glb(state: dict, mesh_simplify: float, texture_size: int) -> Tuple[str, str]:
+    """
+    3D 모델에서 GLB 파일 추출
+    """
     try:
-        # GPU 메모리 정리
-        clear_gpu_memory()
-        
-        # 상태 언패킹
         gs, mesh, trial_id = unpack_state(state)
-        
-        # GLB 변환 전 유효성 검사
-        if gs is None or mesh is None:
-            print("Error: Invalid gaussian or mesh data")
-            return None, None
-            
-        # GLB 변환
-        with torch.inference_mode():
-            try:
-                # 모든 텐서를 CUDA로 이동 (gradient 불필요)
-                device = torch.device('cuda:0')
-                
-                # Gaussian 텐서들을 변환
-                for attr_name in ['_xyz', '_features_dc', '_scaling', '_rotation', '_opacity']:
-                    if hasattr(gs, attr_name):
-                        tensor = getattr(gs, attr_name)
-                        if torch.is_tensor(tensor):
-                            new_tensor = tensor.detach().clone().float().to(device)
-                            setattr(gs, attr_name, new_tensor)
-                
-                # Mesh 텐서들을 변환
-                if hasattr(mesh, 'vertices') and torch.is_tensor(mesh.vertices):
-                    mesh.vertices = mesh.vertices.detach().clone().float().to(device)
-                if hasattr(mesh, 'faces') and torch.is_tensor(mesh.faces):
-                    mesh.faces = mesh.faces.detach().clone().long().to(device)
-                
-                # 추가 속성 확인 및 변환 (gradient 불필요)
-                for attr_name in dir(mesh):
-                    if attr_name.startswith('_'):
-                        continue
-                    attr = getattr(mesh, attr_name)
-                    if torch.is_tensor(attr):
-                        if attr.dtype in [torch.float32, torch.float64]:
-                            setattr(mesh, attr_name, attr.to(device))
-                        else:
-                            setattr(mesh, attr_name, attr.to(device))
-                
-                print("Device and gradient check before GLB conversion:")
-                print(f"Gaussian xyz device: {gs._xyz.device}, requires_grad: {gs._xyz.requires_grad}")
-                print(f"Mesh vertices device: {mesh.vertices.device}, requires_grad: {mesh.vertices.requires_grad}")
-                
-                # GLB 변환
-                glb = postprocessing_utils.to_glb(
-                    gs, 
-                    mesh, 
-                    simplify=mesh_simplify, 
-                    texture_size=texture_size,
-                    verbose=True
-                )
-                
-            except Exception as e:
-                print(f"Error during GLB conversion: {str(e)}")
-                # 디바이스와 gradient 정보 출력
-                if hasattr(gs, '_xyz'):
-                    print(f"Gaussian xyz device: {gs._xyz.device}, requires_grad: {gs._xyz.requires_grad}")
-                if hasattr(mesh, 'vertices'):
-                    print(f"Mesh vertices device: {mesh.vertices.device}, requires_grad: {mesh.vertices.requires_grad}")
-                return None, None
-            
-        if glb is None:
-            print("Error: GLB conversion failed")
-            return None, None
-            
-        # 파일 저장
+        glb = postprocessing_utils.to_glb(gs, mesh, simplify=mesh_simplify, texture_size=texture_size, verbose=False)
         glb_path = f"{TMP_DIR}/{trial_id}.glb"
-        try:
-            glb.export(glb_path)
-            if not os.path.exists(glb_path):
-                print(f"Error: GLB file was not created at {glb_path}")
-                return None, None
-        except Exception as e:
-            print(f"Error saving GLB file: {str(e)}")
-            return None, None
-            
-        print(f"Successfully created GLB file at: {glb_path}")
+        glb.export(glb_path)
         return glb_path, glb_path
-        
     except Exception as e:
-        print(f"Error in extract_glb: {str(e)}")
+        print(f"GLB 추출 중 오류 발생: {e}")
         return None, None
-        
-    finally:
-        # 정리 작업
-        clear_gpu_memory()