From 409b9fbb51c49dda37924e60e589d389d50d69a4 Mon Sep 17 00:00:00 2001 From: henrisel Date: Wed, 17 Dec 2025 15:46:01 +0200 Subject: [PATCH] small cleanup of pipeline script --- 3d-generation-pipeline/generate_model_local.py | 2 +- 3d-generation-pipeline/start_pipeline.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/3d-generation-pipeline/generate_model_local.py b/3d-generation-pipeline/generate_model_local.py index 57ebbe74..aaf6f642 100644 --- a/3d-generation-pipeline/generate_model_local.py +++ b/3d-generation-pipeline/generate_model_local.py @@ -59,7 +59,7 @@ def generate_no_preview(image_base64: str): print(f"Error: {str(e)}") return None -def image_to_3d_api(image_path, output_path): +def image_to_3d_trellis(image_path, output_path): with open(image_path, 'rb') as image_file: image_data = image_file.read() diff --git a/3d-generation-pipeline/start_pipeline.py b/3d-generation-pipeline/start_pipeline.py index a8d7adb4..8c79425f 100644 --- a/3d-generation-pipeline/start_pipeline.py +++ b/3d-generation-pipeline/start_pipeline.py @@ -8,7 +8,7 @@ from datetime import datetime from dotenv import load_dotenv from generate_image_local import text_to_image_invoke_ai -from generate_model_local import image_to_3d_api +from generate_model_local import image_to_3d_trellis load_dotenv() @@ -51,7 +51,7 @@ async def main(): print(f"Generated image file: {image_path}") model_path = pipeline_folder / "models" / timestamp - model_file = image_to_3d_api(image_path, model_path) + model_file = image_to_3d_trellis(image_path, model_path) model_generation_time = time.time() - time_checkpoint logging.info(f"Model generation time: {round(model_generation_time, 1)} s")