Formatting

app.py

@@ -1,5 +1,5 @@
-import torch
 import gradio as gr
+import torch
 from torchaudio.sox_effects import apply_effects_file
 from transformers import AutoFeatureExtractor, AutoModelForAudioXVector
 
@@ -8,7 +8,9 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 STYLE = """
 <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/bootstrap@5.1.3/dist/css/bootstrap.min.css" integrity="sha256-YvdLHPgkqJ8DVUxjjnGVlMMJtNimJ6dYkowFFvp4kKs=" crossorigin="anonymous">
 """
-OUTPUT_OK = STYLE + """
+OUTPUT_OK = (
+    STYLE
+    + """
     <div class="container">
         <div class="row"><h1 style="text-align: center">The speakers are</h1></div>
         <div class="row"><h1 class="display-1 text-success" style="text-align: center">{:.1f}%</h1></div>
@@ -17,7 +19,10 @@ OUTPUT_OK = STYLE + """
         <div class="row"><small style="text-align: center">(You must get at least 85% to be considered the same person)</small><div class="row">
     </div>
 """
-OUTPUT_FAIL = STYLE + """
+)
+OUTPUT_FAIL = (
+    STYLE
+    + """
     <div class="container">
         <div class="row"><h1 style="text-align: center">The speakers are</h1></div>
         <div class="row"><h1 class="display-1 text-danger" style="text-align: center">{:.1f}%</h1></div>
@@ -26,14 +31,15 @@ OUTPUT_FAIL = STYLE + """
         <div class="row"><small style="text-align: center">(You must get at least 85% to be considered the same person)</small><div class="row">
     </div>
 """
+)
 
 EFFECTS = [
-    ['remix', '-'],
+    ["remix", "-"],
     ["channels", "1"],
     ["rate", "16000"],
     ["gain", "-1.0"],
     ["silence", "1", "0.1", "0.1%", "-1", "0.1", "0.1%"],
-    ['trim', '0', '10'],
+    ["trim", "0", "10"],
 ]
 
 THRESHOLD = 0.85
@@ -88,6 +94,27 @@ article = (
     "<a href='https://www.danielpovey.com/files/2018_icassp_xvectors.pdf' target='_blank'>📚 X-Vector paper</a>"
     "</p>"
 )
+examples = [
+    ["samples/cate_blanch.mp3", "samples/cate_blanch_2.mp3"],
+    ["samples/cate_blanch.mp3", "samples/cate_blanch_3.mp3"],
+    ["samples/cate_blanch_2.mp3", "samples/cate_blanch_3.mp3"],
+    ["samples/heath_ledger.mp3", "samples/heath_ledger_2.mp3"],
+    ["samples/heath_ledger.mp3", "samples/heath_ledger_3.mp3"],
+    ["samples/heath_ledger_2.mp3", "samples/heath_ledger_3.mp3"],
+    ["samples/russel_crowe.mp3", "samples/russel_crowe_2.mp3"],
+    ["samples/cate_blanch.mp3", "samples/kirsten_dunst.wav"],
+    ["samples/russel_crowe.mp3", "samples/kirsten_dunst.wav"],
+    ["samples/russel_crowe_2.mp3", "samples/kirsten_dunst.wav"],
+    ["samples/leonardo_dicaprio.mp3", "samples/denzel_washington.mp3"],
+    ["samples/heath_ledger.mp3", "samples/denzel_washington.mp3"],
+    ["samples/heath_ledger_2.mp3", "samples/denzel_washington.mp3"],
+    ["samples/leonardo_dicaprio.mp3", "samples/russel_crowe.mp3"],
+    ["samples/leonardo_dicaprio.mp3", "samples/russel_crowe_2.mp3"],
+    ["samples/naomi_watts.mp3", "samples/denzel_washington.mp3"],
+    ["samples/naomi_watts.mp3", "samples/leonardo_dicaprio.mp3"],
+    ["samples/naomi_watts.mp3", "samples/cate_blanch_2.mp3"],
+    ["samples/naomi_watts.mp3", "samples/kirsten_dunst.wav"],
+]
 
 interface = gr.Interface(
     fn=similarity_fn,
@@ -100,26 +127,6 @@ interface = gr.Interface(
     theme="huggingface",
     allow_flagging=False,
     live=False,
-    examples=[
-        ["samples/cate_blanch.mp3", "samples/cate_blanch_2.mp3"],
-        ["samples/cate_blanch.mp3", "samples/cate_blanch_3.mp3"],
-        ["samples/cate_blanch_2.mp3", "samples/cate_blanch_3.mp3"],
-        ["samples/heath_ledger.mp3", "samples/heath_ledger_2.mp3"],
-        ["samples/heath_ledger.mp3", "samples/heath_ledger_3.mp3"],
-        ["samples/heath_ledger_2.mp3", "samples/heath_ledger_3.mp3"],
-        ["samples/russel_crowe.mp3", "samples/russel_crowe_2.mp3"],
-        ["samples/cate_blanch.mp3", "samples/kirsten_dunst.wav"],
-        ["samples/russel_crowe.mp3", "samples/kirsten_dunst.wav"],
-        ["samples/russel_crowe_2.mp3", "samples/kirsten_dunst.wav"],
-        ["samples/leonardo_dicaprio.mp3", "samples/denzel_washington.mp3"],
-        ["samples/heath_ledger.mp3", "samples/denzel_washington.mp3"],
-        ["samples/heath_ledger_2.mp3", "samples/denzel_washington.mp3"],
-        ["samples/leonardo_dicaprio.mp3", "samples/russel_crowe.mp3"],
-        ["samples/leonardo_dicaprio.mp3", "samples/russel_crowe_2.mp3"],
-        ["samples/naomi_watts.mp3", "samples/denzel_washington.mp3"],
-        ["samples/naomi_watts.mp3", "samples/leonardo_dicaprio.mp3"],
-        ["samples/naomi_watts.mp3", "samples/cate_blanch_2.mp3"],
-        ["samples/naomi_watts.mp3", "samples/kirsten_dunst.wav"],
-    ]
+    examples=examples,
 )
 interface.launch(enable_queue=True)