Upload 3 files

app.py

@@ -7,6 +7,14 @@ from datetime import datetime
 from transformers import WavLMModel
 import torch.nn as nn
 from transformers import AutoTokenizer, AutoModelForCausalLM
+from huggingface_hub import login
+
+# Log in using the Space's secret token if available
+if "HF_TOKEN" in os.environ:
+    print("HF_TOKEN found, logging in...")
+    login(token=os.environ["HF_TOKEN"])
+else:
+    print("WARNING: HF_TOKEN not found in environment variables. Access to gated models may fail.")
 
 print(f"APP STARTUP: {datetime.now()}")
 

requirements.txt

@@ -8,3 +8,4 @@ soundfile>=0.12.0
 librosa>=0.10.0
 accelerate>=0.26.0
 bitsandbytes>=0.41.0
+huggingface_hub>=0.19.0