make 3d model generation model folder loc configurable

3d-generation-pipeline/.env.example

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

3d-generation-pipeline/start_pipeline.py

@@ -2,6 +2,7 @@ import os
 import base64
 import requests
 import argparse
+import subprocess
 from pathlib import Path
 from datetime import datetime
 from dotenv import load_dotenv
@@ -10,6 +11,7 @@ load_dotenv()
 
 ACCOUNT_ID = os.environ["CLOUDFLARE_ACCOUNT_ID"]
 API_TOKEN = os.environ["CLOUDFLARE_API_TOKEN"]
+MODEL_FOLDER = os.environ["MODEL_FOLDER"]
 
 
 def get_timestamp():
@@ -67,11 +69,6 @@ def refine_text_prompt(prompt):
     return data["result"]["response"]
 
 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"
     script_path = MODEL_FOLDER + r"\run.py"