uncomment 1

app.py

@@ -205,7 +205,7 @@ def generate_image(prompt, seed=DEFAULT_SEED,
         progress(1.0, desc="Complete")
         return image, f"Image generated successfully with seed {seed}"
     except Exception as e:
-        print(f"Error in infer: {str(e)}")  
+        print(f"Error in generate_image: {str(e)}")  
         return None, f"Error generating image: {str(e)}"
  
  
@@ -251,44 +251,44 @@ def preload_models():
     return success, status
 
 
-# def pack_state(gs: Gaussian, mesh: MeshExtractResult) -> dict:
-#     return {
-#         'gaussian': {
-#             **gs.init_params,
-#             '_xyz': gs._xyz.cpu().numpy(),
-#             '_features_dc': gs._features_dc.cpu().numpy(),
-#             '_scaling': gs._scaling.cpu().numpy(),
-#             '_rotation': gs._rotation.cpu().numpy(),
-#             '_opacity': gs._opacity.cpu().numpy(),
-#         },
-#         'mesh': {
-#             'vertices': mesh.vertices.cpu().numpy(),
-#             'faces': mesh.faces.cpu().numpy(),
-#         },
-#     }
+def pack_state(gs: Gaussian, mesh: MeshExtractResult) -> dict:
+    return {
+        'gaussian': {
+            **gs.init_params,
+            '_xyz': gs._xyz.cpu().numpy(),
+            '_features_dc': gs._features_dc.cpu().numpy(),
+            '_scaling': gs._scaling.cpu().numpy(),
+            '_rotation': gs._rotation.cpu().numpy(),
+            '_opacity': gs._opacity.cpu().numpy(),
+        },
+        'mesh': {
+            'vertices': mesh.vertices.cpu().numpy(),
+            'faces': mesh.faces.cpu().numpy(),
+        },
+    }
     
     
-# def unpack_state(state: dict) -> Tuple[Gaussian, edict, str]:
-#     gs = Gaussian(
-#         aabb=state['gaussian']['aabb'],
-#         sh_degree=state['gaussian']['sh_degree'],
-#         mininum_kernel_size=state['gaussian']['mininum_kernel_size'],
-#         scaling_bias=state['gaussian']['scaling_bias'],
-#         opacity_bias=state['gaussian']['opacity_bias'],
-#         scaling_activation=state['gaussian']['scaling_activation'],
-#     )
-#     gs._xyz = torch.tensor(state['gaussian']['_xyz'], device='cuda')
-#     gs._features_dc = torch.tensor(state['gaussian']['_features_dc'], device='cuda')
-#     gs._scaling = torch.tensor(state['gaussian']['_scaling'], device='cuda')
-#     gs._rotation = torch.tensor(state['gaussian']['_rotation'], device='cuda')
-#     gs._opacity = torch.tensor(state['gaussian']['_opacity'], device='cuda')
+def unpack_state(state: dict) -> Tuple[Gaussian, edict, str]:
+    gs = Gaussian(
+        aabb=state['gaussian']['aabb'],
+        sh_degree=state['gaussian']['sh_degree'],
+        mininum_kernel_size=state['gaussian']['mininum_kernel_size'],
+        scaling_bias=state['gaussian']['scaling_bias'],
+        opacity_bias=state['gaussian']['opacity_bias'],
+        scaling_activation=state['gaussian']['scaling_activation'],
+    )
+    gs._xyz = torch.tensor(state['gaussian']['_xyz'], device='cuda')
+    gs._features_dc = torch.tensor(state['gaussian']['_features_dc'], device='cuda')
+    gs._scaling = torch.tensor(state['gaussian']['_scaling'], device='cuda')
+    gs._rotation = torch.tensor(state['gaussian']['_rotation'], device='cuda')
+    gs._opacity = torch.tensor(state['gaussian']['_opacity'], device='cuda')
     
-#     mesh = edict(
-#         vertices=torch.tensor(state['mesh']['vertices'], device='cuda'),
-#         faces=torch.tensor(state['mesh']['faces'], device='cuda'),
-#     )
+    mesh = edict(
+        vertices=torch.tensor(state['mesh']['vertices'], device='cuda'),
+        faces=torch.tensor(state['mesh']['faces'], device='cuda'),
+    )
     
-#     return gs, mesh
+    return gs, mesh
 
 
 # @spaces.GPU
@@ -512,7 +512,6 @@ def create_interface():
 
             output_buf = gr.State()
 
-            # Examples section - simplified version that only updates the prompt fields
             gr.Examples(
                 examples=examples,
                 fn=process_example_pipeline,
@@ -521,7 +520,6 @@ def create_interface():
                 cache_examples=True,
             )
 
-        # Event handlers - Fixed to use the renamed components
         gr.on(
             triggers=[prompt_button.click, prompt.submit],
             fn=refine_prompt,