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()
@@ -28,37 +28,37 @@ index = get_index()
 encoder_text = tf.keras.models.load_model("encoder_text_retrievaltext_bmg_221022_54_clean")
 def process(prompt, lang):
     now = datetime.datetime.now()
+    '''
     print()
     print('*************')
     print("Current Time: ", str(now))
     print("Text input : ", prompt)
     print('*************')
     print()
-
-    #try :
-    embed_query  = get_predict(encoder_text, prompt, preprocess_model, model)
-    do_normalize(embed_query)
-    D, I = get_distance(index, embed_query, TOP)
-    print(I)
-    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))
-        
-    return [audio_names[I[0][0]].split('.')[0], get_url_myma(I[0][0], audio_names, url_dict), 
-            audio_names[I[0][1]].split('.')[0], get_url_myma(I[0][1], audio_names, url_dict), 
-            audio_names[I[0][2]].split('.')[0], get_url_myma(I[0][2], audio_names, url_dict), 
-            audio_names[I[0][3]].split('.')[0], get_url_myma(I[0][3], audio_names, url_dict), 
-            audio_names[I[0][4]].split('.')[0], get_url_myma(I[0][4], audio_names, url_dict)]
     '''
+    try :
+        embed_query  = get_predict(encoder_text, prompt, preprocess_model, model)
+        do_normalize(embed_query)
+        D, I = get_distance(index, embed_query, TOP)
+        #print(I)
+        #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))
+            
+        return [audio_names[I[0][0]].split('.')[0], get_url_myma(I[0][0], audio_names, url_dict), 
+                audio_names[I[0][1]].split('.')[0], get_url_myma(I[0][1], audio_names, url_dict), 
+                audio_names[I[0][2]].split('.')[0], get_url_myma(I[0][2], audio_names, url_dict), 
+                audio_names[I[0][3]].split('.')[0], get_url_myma(I[0][3], audio_names, url_dict), 
+                audio_names[I[0][4]].split('.')[0], get_url_myma(I[0][4], audio_names, url_dict)]
     except:
         return ["Error input - please try again", "",
                "Error input - please try again", "",
                "Error input - please try again", "",
                "Error input - please try again", "",
                "Error input - please try again", "",]
-    '''
+    
 inputs = [gr.Textbox(label="Input", value="type your description", max_lines=2), 
             gr.Radio(label="Language", choices=["en"], value="en")]
 
@@ -88,7 +88,7 @@ outputs = [gr.Textbox(label="Track name 1"), gr.Audio(label="Track 1", show_labe
            gr.Textbox(label="Track name 3"), gr.Audio(label="Track 3", show_label=False), 
            gr.Textbox(label="Track name 4"), gr.Audio(label="Track 4", show_label=False), 
            gr.Textbox(label="Track name 5"), gr.Audio(label="Track 5", show_label=False)]
-demo1 = gr.Interface(fn=process, inputs=inputs, outputs=outputs, examples=poc_examples, cache_examples=False, examples_per_page=20)
+demo1 = gr.Interface(fn=process, inputs=inputs, outputs=outputs, examples=poc_examples, cache_examples=True, examples_per_page=20)
 
-demo1.launch(debug=True)
+demo1.launch(debug=False)