Update app.py

app.py

@@ -16,7 +16,7 @@ os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
 # Cacher le nom du repo
 python_path = hf_hub_download(repo_id=os.environ['REPO_ID'], repo_type="space", filename=os.environ['MODEL_FILE'],
                                 use_auth_token=os.environ['TOKEN'])
-print(python_path)
+#print(python_path)
 sys.path.append(os.environ['PRIVATE_DIR'])
 from models import *
 preprocess_model, model = get_models()
@@ -42,11 +42,10 @@ def process(prompt, lang):
     #print(D)
     print("----")
     for i in range(len(I[0])):
-        print(audio_names[I[0][i]], " with distance ", D[0][i])
-        print("    url : ", get_url_myma(I[0][i], audio_names, url_dict))
+        #print(audio_names[I[0][i]], " with distance ", D[0][i])
+        #print("    url : ", get_url_myma(I[0][i], audio_names, url_dict))
         
-    return [get_url_myma(I[0][0], audio_names, url_dict), 
-            audio_names[I[0][0]].split('.')[0],
+    return [audio_names[I[0][0]].split('.')[0], get_url_myma(I[0][0], audio_names, url_dict), 
             get_url_myma(I[0][1], audio_names, url_dict), 
             get_url_myma(I[0][2], audio_names, url_dict), 
             get_url_myma(I[0][3], audio_names, url_dict), 
@@ -76,7 +75,11 @@ poc_examples = [
           ["90 eurodance beat","en"],
 ]
 
-outputs = [gr.Audio(label="Track 1"), gr.Textbox(label="Track name 1"), gr.Audio(label="Track 2"), gr.Audio(label="Track 3"), gr.Audio(label="Track 4"), gr.Audio(label="Track 5")]
+outputs = [gr.Textbox(label="Track name 1"), gr.Audio(label="Track 1"), 
+           gr.Textbox(label="Track name 2"), gr.Audio(label="Track 2"), 
+           gr.Textbox(label="Track name 3"), gr.Audio(label="Track 3"), 
+           gr.Textbox(label="Track name 4"), gr.Audio(label="Track 4"), 
+           gr.Textbox(label="Track name 5"), gr.Audio(label="Track 5")]
 demo1 = gr.Interface(fn=process, inputs=inputs, outputs=outputs, examples=poc_examples, cache_examples=False, examples_per_page=20)
 
 demo1.launch(debug=False)