Update app.py

app.py

@@ -1,6 +1,7 @@
 import io
 import base64
 import torch
+import os
 from flask import Flask, request, jsonify
 from diffusers import StableDiffusionPipeline  # Placeholder; adjust based on SF3D docs
 from PIL import Image
@@ -11,16 +12,17 @@ logger = logging.getLogger(__name__)
 
 app = Flask(__name__)
 
-# Load the model once at startup (on CPU) with token in code
+# Load the model once at startup (on CPU) with token from environment
 try:
     logger.info("Loading Stable Fast 3D pipeline...")
-    # Replace 'your_hf_token_here' with your actual Hugging Face token
-    token = "your_hf_token_here"  # Hardcoded token for authentication
+    token = os.getenv("HF_TOKEN")  # Retrieve token from environment variable
+    if not token:
+        raise ValueError("HF_TOKEN environment variable not set")
     pipe = StableDiffusionPipeline.from_pretrained(
         "stabilityai/stable-fast-3d",
         torch_dtype=torch.float32,
         cache_dir="/tmp/hf_home",
-        token=token,  # Pass token directly
+        token=token,  # Use the environment variable token
     )
     pipe.to("cpu")
     logger.info("=== Application Startup at CPU mode =====")