Update app.py

app.py

@@ -12,14 +12,17 @@ logger = logging.getLogger(__name__)
 
 app = Flask(__name__)
 
-# Load the model once at startup (on CPU) without token (test only)
+# Load the model once at startup (on CPU) with token from environment
 try:
     logger.info("Loading TencentARC InstantMesh pipeline...")
+    token = os.getenv("HF_TOKEN")
+    if not token:
+        raise ValueError("HF_TOKEN environment variable not set")
     pipe = StableDiffusionPipeline.from_pretrained(
         "TencentARC/InstantMesh",
         torch_dtype=torch.float32,
         cache_dir="/tmp/hf_home",
-        # token=token,  # Comment out or remove for test
+        token=token,  # Reintroduce token authentication
     )
     pipe.to("cpu")
     logger.info("=== Application Startup at CPU mode =====")