CleanSong commited on
Commit
af0c9b6
·
verified ·
1 Parent(s): 9999074

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import gradio as gr
2
- from gradio_client import Client
3
  import os
4
 
5
  BACKEND_SPACE = "CleanSong-AI/Main-tool-backend-main"
@@ -11,14 +11,18 @@ def clean_song(file_path):
11
  client = Client(BACKEND_SPACE)
12
 
13
  result = client.predict(
14
- file_path,
15
  api_name="/predict"
16
  )
17
 
18
- print("RAW RESULT FROM BACKEND:", repr(result), type(result))
 
 
19
 
20
- return result
 
21
 
 
22
 
23
  # ------------------------------------------------------------
24
  # UI
 
1
  import gradio as gr
2
+ from gradio_client import Client, handle_file
3
  import os
4
 
5
  BACKEND_SPACE = "CleanSong-AI/Main-tool-backend-main"
 
11
  client = Client(BACKEND_SPACE)
12
 
13
  result = client.predict(
14
+ handle_file(file_path), # ✅ REQUIRED
15
  api_name="/predict"
16
  )
17
 
18
+ # Normalize HF return types
19
+ if isinstance(result, (list, tuple)):
20
+ result = result[0]
21
 
22
+ if isinstance(result, dict) and "name" in result:
23
+ result = result["name"]
24
 
25
+ return result
26
 
27
  # ------------------------------------------------------------
28
  # UI