Update app.py

app.py

@@ -347,29 +347,33 @@ def extract_glb(state: dict, mesh_simplify: float, texture_size: int) -> Tuple[s
         # GLB 변환
         with torch.inference_mode():
             try:
-                # Gaussian 텐서들을 float32로 변환
+                # Gaussian 텐서들을 float32로 변환하고 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):
-                            new_tensor = tensor.clone().detach().float()
+                            # float32로 변환하고 requires_grad 설정
+                            new_tensor = tensor.clone().detach().float().requires_grad_(True)
                             setattr(gs, attr_name, new_tensor)
                 
-                # Mesh 텐서들을 float32로 변환
+                # Mesh 텐서들을 변환
                 if hasattr(mesh, 'vertices') and torch.is_tensor(mesh.vertices):
-                    mesh.vertices = mesh.vertices.clone().detach().float()
+                    mesh.vertices = mesh.vertices.clone().detach().float().requires_grad_(True)
                 if hasattr(mesh, 'faces') and torch.is_tensor(mesh.faces):
-                    mesh.faces = mesh.faces.clone().detach().long()
+                    mesh.faces = mesh.faces.clone().detach().long()  # faces는 정수형 유지
                 
                 # GLB 변환 시도
                 try:
-                    # optimize_uv 매개변수 제거
+                    # 텍스처 베이킹 옵션 조정
                     glb = postprocessing_utils.to_glb(
                         gs, 
                         mesh, 
                         simplify=mesh_simplify, 
                         texture_size=texture_size, 
-                        verbose=True
+                        verbose=True,
+                        bake_texture=True,  # 텍스처 베이킹 활성화
+                        bake_sample_count=100,  # 샘플링 수 조정
+                        bake_resolution=1024  # 베이킹 해상도 설정
                     )
                 except Exception as e:
                     print(f"First GLB conversion attempt failed: {str(e)}")
@@ -378,7 +382,8 @@ def extract_glb(state: dict, mesh_simplify: float, texture_size: int) -> Tuple[s
                         gs, 
                         mesh, 
                         simplify=mesh_simplify, 
-                        texture_size=texture_size
+                        texture_size=texture_size,
+                        bake_texture=False  # 텍스처 베이킹 비활성화
                     )
                 
             except Exception as e: