Update app.py

app.py

@@ -1,21 +1,18 @@
+import gradio as gr
 from gradio_client import Client
 import os
-import gradio as gr
 
-HF_TOKEN = os.environ.get("HF_TOKEN")
+# IMPORTANT: token must be set as env var
+# os.environ["HF_TOKEN"] = "hf_xxx"   # only if not already set
+
 BACKEND_SPACE = "CleanSong-AI/Main-tool-backend-main"
 
 def clean_song(file_path):
     if file_path is None:
         return None
 
-    client = Client(
-        BACKEND_SPACE,
-        hf_token=HF_TOKEN
-    )
-
     try:
-        # This EXACTLY matches how Gradio uploads files
+        client = Client(BACKEND_SPACE)
         result = client.predict(
             file_path,
             api_name="/predict"
@@ -23,17 +20,7 @@ def clean_song(file_path):
     except Exception as e:
         return f"Error calling backend: {e}"
 
-    # result is usually a filepath OR bytes depending on backend
-    if isinstance(result, str) and os.path.exists(result):
-        return result
-
-    # fallback: save bytes
-    output_path = "cleaned_output.wav"
-    with open(output_path, "wb") as f:
-        f.write(result)
-
-    return output_path
-
+    return result
 
 # ------------------------------------------------------------
 # UI