added additional parameters to feature extractor

app.py

@@ -62,7 +62,10 @@ if st.button("Generate Some Music!"):
             else:
                 col4.metric(label=genre['label'], value=f"{genre['score']*100:.2f}%")
 
-    features = classifier.feature_extractor(output)
+    features = classifier.feature_extractor(
+        output, sampling_rate=16_000, return_tensors="pt", padding=True, 
+        return_attention_mask=True, max_length=16_000, truncation=True
+    )
 
     with torch.no_grad():
         vectr = classifier.model(**features, output_hidden_states=True).hidden_states[-1].mean(dim=1)[0]