Add

app.py

@@ -37,7 +37,9 @@ def audio_search(audio_tuple):
                 row["audio"] = [
                     "<audio src=" + orig_row["audio"][0]["src"] + " controls />"
                 ]
-    return pd.DataFrame(rows)[["path", "audio", "sentence", "distance"]].sort_values(by="distance", ascending=True)
+    return pd.DataFrame(rows)[["path", "audio", "sentence", "distance"]].sort_values(
+        by="distance", ascending=True
+    )
 
 
 iface = gr.Interface(
@@ -53,18 +55,18 @@ iface = gr.Interface(
 with gr.Blocks() as demo:
     gr.HTML(
         f"""
-                <h1 style='text-align: center'>
-                <img src="/gradio_api/file=logo.svg" alt="Voice Match" style="width: 95px; height: 30px;"> <br> Voice Match
-                </h1>
-                """
-                )
+        <h1 style='text-align: center; display: flex; align-items: center; justify-content: center;'>
+        <img src="/gradio_api/file=Karaoke_Huggy.png" alt="Voice Match" style="width: 95px; height: 30px; margin-right: 10px"> Voice Match
+        </h1>
+        """
+    )
     gr.HTML(
-        f"""
-                <h2 style='text-align: center'>
-                Powered by <a href="https://huggingface.co/rimelabs/rimecaster">RimeCaster</a>
-                </h2>
-                """
-                )           
+    """
+    <h2 style='text-align: center'>
+    Powered by <a href="https://huggingface.co/rimelabs/rimecaster">RimeCaster</a>
+    </h2>
+    """
+    )
     gr.Markdown(
         f"""
         <div style='text-align: center'>

search.py

@@ -6,6 +6,7 @@ headers = {"Authorization": f"Bearer {os.getenv('HF_TOKEN')}"}
 dataset = "mozilla-foundation/common_voice_17_0"
 config = "en"
 
+
 def _search(paths: list[str]):
     if len(paths) == 0:
         return []
@@ -13,7 +14,6 @@ def _search(paths: list[str]):
         split = "train"
     else:
         split = "validation"
-    
 
     paths_in_clause = ", ".join([f"'{path}'" for path in paths])
     where_clause = f'"path" IN ({paths_in_clause})'
@@ -26,6 +26,7 @@ def _search(paths: list[str]):
 
     return data.get("rows", [])
 
+
 def search(rows: list[dict]):
     file_paths_to_find = [row["path"] for row in rows]
     train_paths = []
@@ -41,8 +42,6 @@ def search(rows: list[dict]):
 
     return train_rows + validation_rows
 
-
-
     paths_in_clause = ", ".join([f"'{path}'" for path in file_paths_to_find])
     where_clause = f'"path" IN ({paths_in_clause})'