Namkoy commited on
Commit
0975118
1 Parent(s): aa013f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from transformers import (
9
  )
10
  from peft import PeftModel, PeftConfig
11
 
12
- os.environ["CUDA_VISIBLE_DEVICES"] = "0"
13
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
14
 
15
  peft_model_id = "Namkoy/whisper_peft_vi_nam"
 
9
  )
10
  from peft import PeftModel, PeftConfig
11
 
12
+ # os.environ["CUDA_VISIBLE_DEVICES"] = "0"
13
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
14
 
15
  peft_model_id = "Namkoy/whisper_peft_vi_nam"