Update app.py

app.py

@@ -8,14 +8,14 @@ import re
 
 processor = Wav2Vec2Processor.from_pretrained("the-cramer-project/Wav2vec-Kyrgyz")
 model = Wav2Vec2ForCTC.from_pretrained("the-cramer-project/Wav2vec-Kyrgyz")
-model.to("cpu")  #.to("cuda")
+model.to("cuda")
 
 def transcribe(file_):
     arr_audio, _ = librosa.load(file_, sr=16000)
     inputs = processor(arr_audio, sampling_rate=16_000, return_tensors="pt", padding=True)
 
     with torch.no_grad():
-        logits = model(inputs.input_values, attention_mask=inputs.attention_mask).logits #model(inputs.input_values.to("cuda"), attention_mask=inputs.attention_mask.to("cuda")).logits
+        logits = model(inputs.input_values.to("cuda"), attention_mask=inputs.attention_mask.to("cuda")).logits
 
     pred_ids = torch.argmax(logits, dim=-1)
     text = processor.batch_decode(pred_ids)[0]