volker commited on
Commit
e929245
·
1 Parent(s): 485e7cf

Use smaller model for transcription.

Browse files
Files changed (1) hide show
  1. basic_agent.py +1 -1
basic_agent.py CHANGED
@@ -80,7 +80,7 @@ class BasicSmolAgent:
80
  " If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."
81
  )
82
  # Load the Whisper pipeline
83
- self.mp3_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-base")
84
 
85
  def get_logs(self):
86
  return self.agent.memory.steps
 
80
  " If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."
81
  )
82
  # Load the Whisper pipeline
83
+ self.mp3_pipe = pipeline("automatic-speech-recognition", model="openai/whisper-tiny")
84
 
85
  def get_logs(self):
86
  return self.agent.memory.steps