Back to CPU

app.py

@@ -8,7 +8,7 @@ model_name = "Yehor/wav2vec2-xls-r-1b-uk-with-lm"
 tokenizer = Wav2Vec2CTCTokenizer.from_pretrained(model_name)
 processor = Wav2Vec2ProcessorWithLM.from_pretrained(model_name)
 model = Wav2Vec2ForCTC.from_pretrained(model_name)
-model.to("cuda")
+model.to("cpu")
 
 
 # define function to read in sound file
@@ -45,7 +45,6 @@ def inference(audio):
         stride_length_s=(4, 2),
         return_tensors="pt",
     ).input_values
-    input_values = input_values.cuda()
 
     with torch.no_grad():
         logits = model(input_values).logits

requirements.txt

@@ -5,8 +5,5 @@ jinja2==3.0.3
 git+https://github.com/huggingface/transformers@bb7949b35a0a2247bc20c7cb6d86764770ca3232
 pyctcdecode==0.3.0
 https://github.com/kpu/kenlm/archive/master.zip
-# torch
--f https://download.pytorch.org/whl/cu113/torch_stable.html
-torch==1.10.2+cu113 
--f https://download.pytorch.org/whl/cu113/torch_stable.html
-torchaudio===0.10.2+cu113 
+torch==1.10.2
+torchaudio===0.10.2