Hugging Face's logo

Spaces:
nvidia
/
audio-flamingo-3
Running on A100

App Files Community
1
goarushi27 commited on
Commit
fa1505c
·
verified ·
1 Parent(s): ac8bd83

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from huggingface_hub import snapshot_download
10
  # ---------------------------------
11
  api_key = os.getenv("my_secret")
12
 
13
- MODEL_BASE_SINGLE = snapshot_download(repo_id="nvidia/audio-flamingo-3", local_dir="./", token=api_key)
14
  MODEL_BASE_THINK = os.path.join(MODEL_BASE_SINGLE, 'stage35')
15
 
16
  model_single = llava.load(MODEL_BASE_SINGLE, model_base=None, devices=[0])
 
10
  # ---------------------------------
11
  api_key = os.getenv("my_secret")
12
 
13
+ MODEL_BASE_SINGLE = snapshot_download(repo_id="nvidia/audio-flamingo-3", token=api_key)
14
  MODEL_BASE_THINK = os.path.join(MODEL_BASE_SINGLE, 'stage35')
15
 
16
  model_single = llava.load(MODEL_BASE_SINGLE, model_base=None, devices=[0])