cgvr/DeltaVR
1
0
Fork 1
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

make 3d model generation model folder loc configurable

Browse Source
This commit is contained in:
Henri Sellis 2025-10-24 10:03:16 +03:00
parent 8b3965d0b1
commit 34a23ab94d
2 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
3d-generation-pipeline/.env.example
View File

@ -1,2 +1,3 @@
CLOUDFLARE_ACCOUNT_ID= CLOUDFLARE_ACCOUNT_ID=
CLOUDFLARE_API_TOKEN= CLOUDFLARE_API_TOKEN=
MODEL_FOLDER=

7
3d-generation-pipeline/start_pipeline.py
View File

@ -2,6 +2,7 @@ import os
import base64 import base64
import requests import requests
import argparse import argparse
import subprocess
from pathlib import Path from pathlib import Path
from datetime import datetime from datetime import datetime
from dotenv import load_dotenv from dotenv import load_dotenv
@ -10,6 +11,7 @@ load_dotenv()
ACCOUNT_ID = os.environ["CLOUDFLARE_ACCOUNT_ID"] ACCOUNT_ID = os.environ["CLOUDFLARE_ACCOUNT_ID"]
API_TOKEN = os.environ["CLOUDFLARE_API_TOKEN"] API_TOKEN = os.environ["CLOUDFLARE_API_TOKEN"]
MODEL_FOLDER = os.environ["MODEL_FOLDER"]
def get_timestamp(): def get_timestamp():
@ -67,11 +69,6 @@ def refine_text_prompt(prompt):
return data["result"]["response"] return data["result"]["response"]
def image_to_3d(image_path, output_path): def image_to_3d(image_path, output_path):
import subprocess
# TODO: make this configurable
MODEL_FOLDER = r"D:\henrisel\AI-models\stable-fast-3d"
venv_python = MODEL_FOLDER + r"\.venv\Scripts\python.exe" venv_python = MODEL_FOLDER + r"\.venv\Scripts\python.exe"
script_path = MODEL_FOLDER + r"\run.py" script_path = MODEL_FOLDER + r"\run.py"