token

app.py

@@ -19,6 +19,7 @@ from trellis.pipelines import TrellisImageTo3DPipeline
 from trellis.representations import Gaussian, MeshExtractResult
 from trellis.utils import render_utils, postprocessing_utils
 
+huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
 # Constants
 MAX_SEED = np.iinfo(np.int32).max
 TMP_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'tmp')
@@ -273,9 +274,11 @@ with gr.Blocks() as demo:
 if __name__ == "__main__":
     # Initialize Flux pipeline
     device = "cuda" if torch.cuda.is_available() else "cpu"
+    huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
     flux_pipeline = DiffusionPipeline.from_pretrained(
         "black-forest-labs/FLUX.1-dev", 
-        torch_dtype=torch.bfloat16
+        torch_dtype=torch.bfloat16,
+        token = huggingface_token
     )
     flux_pipeline.load_lora_weights("gokaygokay/Flux-Game-Assets-LoRA-v2")
     flux_pipeline = flux_pipeline.to(device)