Update app.py

app.py

@@ -1,16 +1,18 @@
 import gradio as gr
 import torch
+import librosa
 from transformers import AutoModelForAudioClassification, AutoFeatureExtractor
 
 # Load model directly from HF Hub
 repo_name = "m7k-run/ast-genre-classifier"
 model = AutoModelForAudioClassification.from_pretrained(repo_name)
 feature_extractor = AutoFeatureExtractor.from_pretrained(repo_name)
-
 def predict(audio):
+    audio_array, _ = librosa.load(audio, sr=16000)  
+    
     inputs = feature_extractor(
-        audio["array"],
-        sampling_rate=audio["sampling_rate"],
+        audio_array,
+        sampling_rate=16000,
         return_tensors="pt"
     )
     
@@ -23,7 +25,7 @@ def predict(audio):
 
 demo = gr.Interface(
     fn=predict,
-    inputs=gr.Audio(type="numpy"),  
+    inputs=gr.Audio(type="filepath"),  
     outputs=gr.Label(),
     title="Audio Classifier"
 )