Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,6 +8,7 @@ from transformers import (
|
|
| 8 |
)
|
| 9 |
from peft import PeftModel, PeftConfig
|
| 10 |
|
|
|
|
| 11 |
|
| 12 |
peft_model_id = "Namkoy/whisper_peft_vi_nam"
|
| 13 |
language = "vietnamese"
|
|
|
|
| 8 |
)
|
| 9 |
from peft import PeftModel, PeftConfig
|
| 10 |
|
| 11 |
+
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
|
| 12 |
|
| 13 |
peft_model_id = "Namkoy/whisper_peft_vi_nam"
|
| 14 |
language = "vietnamese"
|