Update player.py

player.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from huggingface_hub import HfApi, list_files_info
 token = ""
+source_dataset = "jbilcke-hf/ai-tube-index"
 api = HfApi(token=token)
 
 
@@ -16,9 +17,14 @@ def find_dataset(author=None):
             spaces.append(f'{space.id}<br>')
             out_html = f'{out_html}<b>{space.id}</b><br>'
             file_box = [info.path for info in list_files_info(f'{space.id}', repo_type="dataset")]
+            ai_tube_box = [info.path for info in list_files_info(f'{source_dataset}', repo_type="dataset")]
             for file in file_box:
                 if file.startswith("prompt_"):
+                    prompt_file = file
                     out_html=f'{out_html}  • {file}<br>'
+                    if prompt_file.split("prompt_",1)[1].split(".",1)[0] in file_box:
+                        print("yes")
+                    
             #print([info.path for info in list_files_info(f'{space.id}', repo_type="dataset")])
             #print(space.id)
         except Exception as e: