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 [email protected] |
-'''.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)