Samarth991 commited on
Commit
6a3f9b4
1 Parent(s): 3dc3903

whisper llama

Browse files
Files changed (1) hide show
  1. whisper_app.py +1 -1
whisper_app.py CHANGED
@@ -40,7 +40,6 @@ class WHISPERModel:
40
  raise("Unable to reach for URL {}".format(audio_path))
41
  else:
42
  audio = self.read_audio(audio_path)
43
- if audio :
44
  conv_language = self.get_info(audio)
45
  if conv_language !='en':
46
  res = self.model.transcribe(audio,task='translate')
@@ -50,6 +49,7 @@ class WHISPERModel:
50
  res = self.model.transcribe(audio)
51
  audio_duration = audio.shape[0] / SAMPLE_RATE
52
  text_data['text'] = res['text']
 
53
  text_data['duration'] = audio_duration
54
  text_data['language'] = conv_language
55
  return text_data
 
40
  raise("Unable to reach for URL {}".format(audio_path))
41
  else:
42
  audio = self.read_audio(audio_path)
 
43
  conv_language = self.get_info(audio)
44
  if conv_language !='en':
45
  res = self.model.transcribe(audio,task='translate')
 
49
  res = self.model.transcribe(audio)
50
  audio_duration = audio.shape[0] / SAMPLE_RATE
51
  text_data['text'] = res['text']
52
+
53
  text_data['duration'] = audio_duration
54
  text_data['language'] = conv_language
55
  return text_data