ruffed

.gitignore

@@ -1,3 +1,5 @@
 .idea/
 .venv/
-flagged/
+.ruff_cache/
+
+flagged/

app.py

@@ -2,7 +2,6 @@ import sys
 import time
 
 import torch
-import torchaudio
 import librosa
 
 import gradio as gr
@@ -21,7 +20,9 @@ concurrency_limit = 1
 use_torch_compile = False
 
 # Load the model
-asr_model = AutoModelForCTC.from_pretrained(model_name, torch_dtype=torch_dtype).to(device)
+asr_model = AutoModelForCTC.from_pretrained(model_name, torch_dtype=torch_dtype).to(
+    device
+)
 processor = Wav2Vec2BertProcessor.from_pretrained(model_name)
 
 if use_torch_compile:
@@ -37,7 +38,7 @@ examples = [
     "example_6.wav",
 ]
 
-examples_table = '''
+examples_table = """
 | File  | Text |
 | ------------- | ------------- |
 | `example_1.wav`  | тема про яку не люблять говорити офіційні джерела у генштабі і міноборони це хімічна зброя окупанти вже тривалий час використовують хімічну зброю заборонену |
@@ -46,10 +47,10 @@ examples_table = '''
 | `example_4.wav`  | використовує на фронті все що має і хімічна зброя не нийняток тож з чим маємо справу розбиралася марія моганисян |
 | `example_5.wav`  | двох тисяч випадків застосування росіянами боєприпасів споряджених небезпечними хімічними речовинами |
 | `example_6.wav`  | на всі писані норми марія моганисян олександр моторний спецкор марафон єдині новини |
-'''.strip()
+""".strip()
 
 # https://www.tablesgenerator.com/markdown_tables
-authors_table = '''
+authors_table = """
 ## Authors
 
 Follow them in social networks and **contact** if you need any help or have any questions:
@@ -61,7 +62,7 @@ Follow them in social networks and **contact** if you need any help or have any
 | https://github.com/egorsmkv at GitHub                                                           |
 | https://huggingface.co/Yehor at Hugging Face                                                    |
 | or use egorsmkv@gmail.com                                                                       |
-'''.strip()
+""".strip()
 
 description_head = f"""
 # Speech-to-Text for Ukrainian v2
@@ -121,7 +122,9 @@ def inference(audio_path, progress=gr.Progress()):
 
     duration = librosa.get_duration(path=audio_path)
     if duration < min_duration:
-        raise gr.Error(f"The duration of the file is less than {min_duration} seconds, it is {round(duration, 2)} seconds.")
+        raise gr.Error(
+            f"The duration of the file is less than {min_duration} seconds, it is {round(duration, 2)} seconds."
+        )
     if duration > max_duration:
         raise gr.Error(f"The duration of the file exceeds {max_duration} seconds.")
 
@@ -150,7 +153,7 @@ def inference(audio_path, progress=gr.Progress()):
         predictions = processor.batch_decode(predicted_ids)
 
         if not predictions:
-            predictions = '-'
+            predictions = "-"
 
         elapsed_time = round(time.time() - t0, 2)
         rtf = round(elapsed_time / audio_duration, 4)
@@ -207,12 +210,12 @@ with demo:
 
     with gr.Row():
         gr.Examples(label="Choose an example", inputs=audio_file, examples=examples)
-    
+
     gr.Markdown(examples_table)
 
     gr.Markdown(description_foot)
 
-    gr.Markdown('### Gradio app uses the following technologies:')
+    gr.Markdown("### Gradio app uses the following technologies:")
     with gr.Row():
         gr.Markdown(tech_env)
         gr.Markdown(tech_libraries)