Update app.py

app.py

@@ -8,9 +8,8 @@ from huggingface_hub import snapshot_download
 # ---------------------------------
 # SINGLE-TURN MODEL SETUP
 # ---------------------------------
-api_key = os.getenv("my_secret")
 
-MODEL_BASE_SINGLE = snapshot_download(repo_id="nvidia/audio-flamingo-3", token=api_key)
+MODEL_BASE_SINGLE = snapshot_download(repo_id="nvidia/audio-flamingo-3")
 MODEL_BASE_THINK = os.path.join(MODEL_BASE_SINGLE, 'stage35')
 
 model_single = llava.load(MODEL_BASE_SINGLE, model_base=None, devices=[0])
@@ -26,7 +25,7 @@ model_think = PeftModel.from_pretrained(
 # ---------------------------------
 # MULTI-TURN MODEL SETUP
 # ---------------------------------
-MODEL_BASE_MULTI = "nvidia/audio-flamingo-3-chat"
+MODEL_BASE_MULTI = snapshot_download(repo_id="nvidia/audio-flamingo-3-chat")
 model_multi = llava.load(MODEL_BASE_MULTI, model_base=None, devices=[0])
 generation_config_multi = model_multi.default_generation_config