Update app.py

app.py

@@ -2,15 +2,16 @@ import gradio as gr
 from transformers import AutoProcessor, AutoModel
 import torch
 import torchaudio
+import os
 
-# Load the processor and model
-processor = AutoProcessor.from_pretrained("google/hear-pytorch")
-model = AutoModel.from_pretrained("google/hear-pytorch")
+HF_TOKEN = os.getenv("hear2")
+
+processor = AutoProcessor.from_pretrained("google/hear-pytorch", use_auth_token=HF_TOKEN)
+model = AutoModel.from_pretrained("google/hear-pytorch", use_auth_token=HF_TOKEN)
 
 def predict(audio):
     waveform, sample_rate = torchaudio.load(audio)
     inputs = processor(waveform, sampling_rate=sample_rate, return_tensors="pt")
-
     with torch.no_grad():
         embeddings = model(**inputs).last_hidden_state
     return f"Embedding shape: {embeddings.shape}"