add possibility of truncating

app.py

@@ -7,6 +7,14 @@ model_id = f"{username}/distilhubert-finetuned-gtzan"
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 pipe = pipeline("audio-classification", model=model_id, device=device)
 
+# def predict_trunc(filepath):
+#     preprocessed = pipe.preprocess(filepath)
+#     truncated = pipe.feature_extractor.pad(preprocessed,truncation=True, max_length = 16_000*30)
+#     model_outputs = pipe.forward(truncated)
+#     outputs = pipe.postprocess(model_outputs)
+
+#     return outputs
+
 
 def classify_audio(filepath):
     """
@@ -18,6 +26,7 @@ def classify_audio(filepath):
      {"country":  0.8339303731918335, "rock":0.11914275586605072}
     """
     preds = pipe(filepath)
+    # preds = predict_trunc(filepath)
     outputs = {}
     for p in preds:
         outputs[p["label"]] = p["score"]