image

app.py

@@ -143,8 +143,8 @@ from audio_separator import Separator
 
 def inference(audio, vocals, bass, drums, other, piano, guitar, lead_vocals, backing_vocals):
     # Initially, show the loading GIF
-    loading_gif_path = "./7RwF.gif"  
-    transparent_img_path = "./images.png" 
+    loading_gif_path = "7RwF.gif"
+    gr.Image(loading_gif_path,visible=True) 
 
     os.makedirs("out", exist_ok=True)
     audio_path = 'test.wav'
@@ -179,7 +179,8 @@ def inference(audio, vocals, bass, drums, other, piano, guitar, lead_vocals, bac
     }
 
     # Once processing is done, hide the GIF by returning a transparent image
-    return [gr.Audio(stem_paths[stem], visible=bool(stem_paths[stem])) for stem in stem_paths] + [transparent_img_path]
+    gr.Image(visible=False)
+    return [gr.Audio(stem_paths[stem], visible=bool(stem_paths[stem])) for stem in stem_paths]  
 
 # Define checkboxes for each stem
 checkbox_labels = ["Full Vocals", "Bass", "Drums", "Other", "Piano", "Guitar", "Lead Vocals", "Backing Vocals"]
@@ -191,7 +192,7 @@ description = "Music Source Separation in the Waveform Domain. Upload your audio
 iface = gr.Interface(
     inference, 
     [gr.components.Audio(type="numpy", label="Input")] + checkboxes,
-    [gr.Audio(label=label, visible=False) for label in checkbox_labels] + [gr.Image()],
+    [gr.Audio(label=label, visible=False) for label in checkbox_labels],
     title=title,
     description=description,
 )