Spaces:
Sleeping
Sleeping
Yehor Smoliakov
commited on
Commit
·
97f41b1
1
Parent(s):
774c0b8
Add theme
Browse files
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:
|