Yehor Smoliakov commited on
Commit
97f41b1
·
1 Parent(s): 774c0b8
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -10,7 +10,8 @@ from transformers import AutoModelForCTC, Wav2Vec2BertProcessor
10
 
11
  # Config
12
  model_name = "Yehor/w2v-bert-2.0-uk-v2"
13
- device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
14
  torch_dtype = torch.float16
15
 
16
  min_duration = 0.5
@@ -187,6 +188,7 @@ def inference(audio_path, progress=gr.Progress()):
187
  demo = gr.Blocks(
188
  title="Speech-to-Text for Ukrainian",
189
  analytics_enabled=False,
 
190
  )
191
 
192
  with demo:
 
10
 
11
  # Config
12
  model_name = "Yehor/w2v-bert-2.0-uk-v2"
13
+ # device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
14
+ device = 'cpu'
15
  torch_dtype = torch.float16
16
 
17
  min_duration = 0.5
 
188
  demo = gr.Blocks(
189
  title="Speech-to-Text for Ukrainian",
190
  analytics_enabled=False,
191
+ theme=gr.themes.Base(),
192
  )
193
 
194
  with demo: