1
0
forked from cgvr/DeltaVR

small cleanup of pipeline script

This commit is contained in:
2025-12-17 15:46:01 +02:00
parent 2c19602e9b
commit 409b9fbb51
2 changed files with 3 additions and 3 deletions

View File

@@ -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()

View File

@@ -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")