Add huggingface login and set code directory

app.py

@@ -1,6 +1,19 @@
 from dataclasses import dataclass, field
 import logging
 
+import sys
+sys.path.append("/home/user/app/src/sonicverse")
+
+from huggingface_hub import login
+import os
+
+# Ensure this runs early in your app
+hf_token = os.getenv("HUGGINGFACE_HUB_TOKEN")
+if not hf_token:
+    raise ValueError("Missing HUGGINGFACE_HUB_TOKEN. Set it as a secret in your Space.")
+
+login(token=hf_token)
+
 import gradio as gr
 import torch
 import transformers
@@ -35,13 +48,13 @@ model, tokenizer = load_trained_lora_model(
 
 
 def generate_caption(audio_file):
-    waveform, sample_rate = torchaudio.load(audio_file)
+    # waveform, sample_rate = torchaudio.load(audio_file)
 
     req_json = {
-        "audio": {
-            "tensor": waveform,
-            "sampling_rate": sample_rate,
-        }
+        "messages": [
+            {"role": "user", "content": "Describe the music. <sound>"}
+        ],
+        "sounds": [audio_file]
     }
 
     encoded_dict = encode_chat(req_json, tokenizer, model.modalities)