Update app.py

app.py

@@ -347,14 +347,17 @@ def extract_glb(state: dict, mesh_simplify: float, texture_size: int) -> Tuple[s
         # GLB 변환
         with torch.inference_mode():
             try:
-                # 텐서들의 requires_grad를 False로 설정
-                for param in gs.parameters():
-                    if hasattr(param, 'requires_grad'):
-                        param.requires_grad_(False)
+                # Gaussian 객체의 텐서들에 대해 requires_grad 설정
+                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):
+                            tensor.requires_grad_(False)
                 
-                if hasattr(mesh, 'vertices'):
+                # Mesh 텐서들에 대해 requires_grad 설정
+                if hasattr(mesh, 'vertices') and torch.is_tensor(mesh.vertices):
                     mesh.vertices.requires_grad_(False)
-                if hasattr(mesh, 'faces'):
+                if hasattr(mesh, 'faces') and torch.is_tensor(mesh.faces):
                     mesh.faces.requires_grad_(False)
                 
                 glb = postprocessing_utils.to_glb(