Update app.py

app.py

@@ -10,7 +10,7 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 model_card = ModelCard(
     name="S-KEY Detection",
     description="Musical key detection using Self-supervised Key Detection.",
-    author="Deezer Research",
+    author="Yuexuan Kong et al.",
     tags=["key detection", "mir"],
 )
 
@@ -18,7 +18,12 @@ model_card = ModelCard(
 def process_fn(input_audio_path):
     try:
         ext = input_audio_path.split(".")[-1].lower()
-        detected_key = detect_key(input_audio_path, ext, DEVICE)
+        
+        detected_key = detect_key(
+            audio_dir=input_audio_path,
+            extension=ext,
+            device=DEVICE
+        )
 
         if isinstance(detected_key, list):
             detected_key = detected_key[0]