Update model.py

model.py

@@ -7,10 +7,14 @@ from io import BytesIO
 
 def load_pipeline():
     """
-    Load the stable-zero123 model pipeline from Hugging Face.
+    Load the LGM-full model pipeline from Hugging Face with remote code execution enabled.
     """
     ckpt_id = "dylanebert/LGM-full"
-    pipe = DiffusionPipeline.from_pretrained(ckpt_id, torch_dtype=torch.float32).to("cpu")
+    pipe = DiffusionPipeline.from_pretrained(
+        ckpt_id, 
+        torch_dtype=torch.float32, 
+        trust_remote_code=True  # Enable remote code execution for custom model code
+    ).to("cpu")
     return pipe
 
 def generate_3d_model(pipe, prompt, output_path="output.obj", guidance_scale=7.5, num_inference_steps=32):