Spaces:
Running
on
Zero
Running
on
Zero
github-actions[bot]
commited on
Commit
·
9e6abf1
1
Parent(s):
d0aba40
Sync with https://github.com/mozilla-ai/speech-to-text-finetune
Browse files
app.py
CHANGED
|
@@ -53,6 +53,7 @@ def _load_hf_model(model_repo_id: str) -> Tuple[Pipeline | None, str]:
|
|
| 53 |
return pipeline(
|
| 54 |
"automatic-speech-recognition",
|
| 55 |
model=model_repo_id,
|
|
|
|
| 56 |
), f"✅ HF Model {model_repo_id} has been loaded."
|
| 57 |
|
| 58 |
|
|
|
|
| 53 |
return pipeline(
|
| 54 |
"automatic-speech-recognition",
|
| 55 |
model=model_repo_id,
|
| 56 |
+
device=0
|
| 57 |
), f"✅ HF Model {model_repo_id} has been loaded."
|
| 58 |
|
| 59 |
|