Update app.py

app.py

@@ -8,7 +8,9 @@ from huggingface_hub import snapshot_download
 # ---------------------------------
 # SINGLE-TURN MODEL SETUP
 # ---------------------------------
-MODEL_BASE_SINGLE = snapshot_download("nvidia/audio-flamingo-3")
+api_key = os.getenv("my_secret")
+
+MODEL_BASE_SINGLE = snapshot_download(repo_id="nvidia/audio-flamingo-3", local_dir="./", token=api_key)
 MODEL_BASE_THINK = os.path.join(MODEL_BASE_SINGLE, 'stage35')
 
 model_single = llava.load(MODEL_BASE_SINGLE, model_base=None, devices=[0])
@@ -83,11 +85,12 @@ def multi_turn_chat(user_input, audio_file, history, current_audio):
 # ---------------------------------
 # INTERFACE
 # ---------------------------------
-with gr.Blocks(css=".gradio-container { max-width: 950px !important; margin: auto; }") as demo:
+with gr.Blocks(css=".gradio-container { max-width: 100% !important; margin: 0 auto !important; padding: 0 2rem; }") as demo:
     gr.Markdown("""
 
-
 <div align="center">
+  <img src="https://github.com/NVIDIA/audio-flamingo/blob/audio_flamingo_3/static/logo-no-bg.png" alt="Audio Flamingo 3 Logo" width="120" style="margin-bottom: 10px;">
+  
   <h2><strong>Audio Flamingo 3</strong></h2>
   <p><em>Advancing Audio Intelligence with Fully Open Large Audio-Language Models</em></p>
 </div>