Update app.py

app.py

@@ -8,12 +8,12 @@ import re
 
 processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
 model = Wav2Vec2ForCTC.from_pretrained("facebook/wav2vec2-base-960h")
-model.to("cuda")
+model.to("cpu")  #.to("cuda")
 
 def transcribe(file_):
     arr_audio, _ = librosa.load(file_, sr=16000)
     input_values = processor(arr_audio, return_tensors="pt", padding="longest").input_values
-    logits = model(input_values.to("cuda")).logits
+    logits = model(input_values).logits  #model(input_values.to("cuda")).logits
     predicted_ids = torch.argmax(logits, dim=-1)
     transcription = processor.batch_decode(predicted_ids)
     return transcription[0].lower()