Update TangoFlux.py

TangoFlux.py

@@ -19,16 +19,17 @@ import json
 import inspect
 import yaml
 from safetensors.torch import load_file
-
+import os
+print(os.environ['HOME'])
 
 class TangoFluxInference:
 
     def __init__(self,name='declare-lab/TangoFlux',device="cuda"):
         
 
-        self.vae = AutoencoderOobleck.from_pretrained("stabilityai/stable-audio-open-1.0",subfolder='vae')
+        self.vae = AutoencoderOobleck.from_pretrained("stabilityai/stable-audio-open-1.0",subfolder='vae',token=os.environ['HF_TOKEN'])
 
-        paths = snapshot_download(repo_id=name) 
+        paths = snapshot_download(repo_id=name,token=os.environ['HF_TOKEN']) 
         weights = load_file("{}/tangoflux.safetensors".format(paths))
 
         with open('{}/config.json'.format(paths),'r') as f: