Update app.py

app.py

@@ -18,13 +18,7 @@ embed_html2 = '" title="YouTube video player" frameborder="0" allow="acceleromet
 os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
 
-# Cacher le nom du repo
-
-#python_path = hf_hub_download(repo_id='PierreHanna/AudioSimilarity', repo_type="space", filename='models.py',
-#                                use_auth_token='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')
-#python_path = hf_hub_download(repo_id='PierreHanna/AudioSimilarity', repo_type="space", filename=os.environ['MODEL_FILE'],
-#                                use_auth_token=os.environ['TOKEN'])
-
+#
 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)
@@ -54,32 +48,11 @@ def process_file(input_path):
     return process(input_path, '')
 
 def process(audio_file, embed_html_all):
-    #model = openl3.models.load_audio_embedding_model(input_repr="linear", content_type="music",embedding_size=512)
-    #audio, sr = librosa.load(audio_file, mono=True)
-    #emb, ts = openl3.get_audio_embedding(audio, sr, model=model)  
-    #emb = np.mean(emb, axis=0)
-    #print("SHAPE ", emb.shape)
-
     emb, ts = get_embed(audio_file)
-    
-    #ind = faiss.read_index("index.index")
-    #ind_filenames = pickle.load(open('index.filenames','rb'))
     ind, ind_filenames = load_index()
-    
-    #dict_bmg = pickle.load(open('dict_bmg.pickle','rb')) # filename to url
     catalog = load_catalog()
-
     D, I = do_search(emb, ind)
-    #D, I = ind.search(emb.reshape((1,512)), 5) 
-
-    #top1 = dict_bmg[ind_filenames[I[0][0]]]
-    #top2 = dict_bmg[ind_filenames[I[0][1]]]
-    #top3 = dict_bmg[ind_filenames[I[0][2]]]
-    #top4 = dict_bmg[ind_filenames[I[0][3]]]
-    #top5 = dict_bmg[ind_filenames[I[0][4]]]
-
-    top1, top2, top3, top4, top5 = get_top(I, ind_filenames, catalog)
-    
+    top1, top2, top3, top4, top5 = get_top(I, ind_filenames, catalog)   
     return top1, top2, top3, top4, top5