Add HF_TOKEN authentication support for model access

README.md

@@ -31,6 +31,17 @@ Generate up to 4 audio variations from a single text prompt using Stability AI's
 - **Variable Duration**: Control audio length (1-11 seconds)
 - **Fast Generation**: Uses optimized pingpong sampler with 8 steps
 
+## Setup
+
+This model requires accepting the license agreement on Hugging Face. To use this Space:
+
+1. **Accept the model license**: Visit [stabilityai/stable-audio-open-small](https://huggingface.co/stabilityai/stable-audio-open-small) and accept the license agreement
+2. **Create an access token**: Go to [Settings > Access Tokens](https://huggingface.co/settings/tokens) and create a token with "read" permissions
+3. **Add token to Space**: In your Space settings, go to "Variables and secrets" and add a new secret:
+   - Name: `HF_TOKEN`
+   - Value: Your access token
+   - Make sure it's marked as private
+
 ## Usage
 
 1. Enter a text prompt describing the audio you want to generate

app.py

@@ -1,8 +1,10 @@
 import torch
 import torchaudio
 import gradio as gr
+import os
 from stable_audio_tools import get_pretrained_model
 from stable_audio_tools.inference.generation import generate_diffusion_cond
+from huggingface_hub import login
 
 # Global model variables
 model = None
@@ -16,6 +18,15 @@ def load_model():
     device = "cuda" if torch.cuda.is_available() else "cpu"
     print(f"Loading model on device: {device}")
     
+    # Check for HF_TOKEN environment variable (set in Space settings)
+    hf_token = os.getenv("HF_TOKEN")
+    if hf_token:
+        print("Using HF_TOKEN for authentication")
+        login(token=hf_token)
+    else:
+        print("Warning: HF_TOKEN not found. Model access may fail if authentication is required.")
+        print("Please set HF_TOKEN as a secret in your Space settings.")
+    
     # Download and load the pretrained model
     model, model_config = get_pretrained_model("stabilityai/stable-audio-open-small")
     sample_rate = model_config["sample_rate"]
@@ -99,6 +110,8 @@ with gr.Blocks(title="Stable Audio Open Small - 4 Variations") as demo:
     
     **Model**: [stabilityai/stable-audio-open-small](https://huggingface.co/stabilityai/stable-audio-open-small)
     
+    **Note**: This model requires accepting the license agreement. Make sure to set `HF_TOKEN` as a secret in your Space settings.
+    
     Enter a text description and click Generate to create 4 different audio variations.
     """)