Beijuka commited on
Commit
4564153
·
verified ·
1 Parent(s): 51d0d81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -35,9 +35,9 @@ model_map = {
35
  os.makedirs("responses", exist_ok=True)
36
 
37
  # Transcription function
38
- device = 0 if torch.cuda.is_available() else -1
39
  def transcribe(audio, language):
40
- asr = pipeline("automatic-speech-recognition", model=model_map[language], device=device, token=HF_TOKEN)
41
  text = asr(audio)["text"]
42
  return text, audio
43
 
 
35
  os.makedirs("responses", exist_ok=True)
36
 
37
  # Transcription function
38
+ inference_device = 0 if torch.cuda.is_available() else -1
39
  def transcribe(audio, language):
40
+ asr = pipeline("automatic-speech-recognition", model=model_map[language], device=inference_device, token=HF_TOKEN)
41
  text = asr(audio)["text"]
42
  return text, audio
43