Upload app.py

app.py

@@ -10,7 +10,6 @@ description = """
 <a style='display:inline-block' href='https://ai-muzic.github.io/clamp/'><img src='https://img.shields.io/badge/website-CLaMP-ff69b4.svg' /></a>
 <a style="display:inline-block" href="https://huggingface.co/datasets/sander-wood/wikimusictext"><img src="https://img.shields.io/badge/huggingface-dataset-ffcc66.svg"></a>
 <a style="display:inline-block" href="https://arxiv.org/pdf/2304.11029.pdf"><img src="https://img.shields.io/badge/arXiv-2304.11029-b31b1b.svg"></a>
-<a class="duplicate-button" style="display:inline-block" target="_blank" href="https://huggingface.co/spaces/sander-wood/clamp_semantic_music_search?duplicate=true"><img style="margin-top:0;margin-bottom:0" src="https://huggingface.co/datasets/huggingface/badges/raw/main/duplicate-this-space-md-dark.svg" alt="Duplicate Space"></a>
 </div>
 
 ## ℹ️ How to use this demo?
@@ -150,7 +149,7 @@ def semantic_music_search(query):
     """
     with open(KEY_MODAL+"_key_cache_"+str(music_length)+".pth", 'rb') as f:
         key_cache = torch.load(f)
-        
+    print("\nQuery: "+query+"\n")
     # encode query
     query_ids = encoding_data([unidecode(query)], QUERY_MODAL)
     query_feature = get_features(query_ids, QUERY_MODAL)
@@ -173,6 +172,11 @@ def semantic_music_search(query):
             # output += "Genre:\n" + item['genre']+ '\n\n'
             # output += "Description:\n" + item['text']+ '\n\n'
             # output += "ABC notation:\n" + item['music']+ '\n\n'
+            print("Title:\n" + item['title'])
+            print("Artist:\n" + item['artist'])
+            print("Genre:\n" + item['genre'])
+            print("Description:\n" + item['text'])
+            print("ABC notation:\n" + item['music'])
             return item["title"], item["artist"], item["genre"], item["text"], item["music"]
 
 output_title = gr.outputs.Textbox(label="Title")