Namkoy commited on
Commit
9847e36
1 Parent(s): 378382b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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"