forked from cgvr/DeltaVR
small cleanup of pipeline script
This commit is contained in:
@@ -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()
|
||||
|
||||
|
||||
@@ -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")
|
||||
|
||||
Reference in New Issue
Block a user