Update app.py

app.py

@@ -9,8 +9,8 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 # Create a ModelCard
 model_card = ModelCard(
     name="S-KEY Detection",
-    description="Musical key detection using Self-supervised Key Detection.",
-    author="Yuexuan Kong et al.",
+    description="Major/Minor Key detection from audio using Self-supervised Key Detection.",
+    author="Deezer Research",
     tags=["key detection", "mir"],
 )
 
@@ -18,13 +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
-        )
+        #result = detect_key(audio_path=input_audio_path, extension=ext, device=DEVICE,)
+        #return result
 
+        # Ensure detected_key is a string
         if isinstance(detected_key, list):
             detected_key = detected_key[0]
 
@@ -49,6 +48,7 @@ def process_fn(input_audio_path):
 # Build Gradio Endpoint
 with gr.Blocks() as demo:
     input_audio = gr.Audio(label="Input Audio", type="filepath").harp_required(True)
+    # output_text = gr.Textbox(label="Detected Key")
     output_audio = gr.Audio(label="Output Audio", type="filepath")
     output_labels = gr.JSON(label="Analysis Labels")
 
@@ -61,4 +61,4 @@ with gr.Blocks() as demo:
 
 
 if __name__ == "__main__":
-    demo.queue().launch()
+    demo.queue().launch()