arjunanand13 commited on
Commit
9f6ee8b
1 Parent(s): 4666d81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +22 -11
app.py CHANGED
@@ -175,18 +175,29 @@ class VideoClassifier:
175
  video_name = mp4_file.split("/")[-1]
176
  wav_file = "results/audiotrack.wav"
177
  video_clip = VideoFileClip(mp4_file)
178
- audioclip = video_clip.audio
179
- wav_file = audioclip.write_audiofile(wav_file)
180
- audioclip.close()
181
  video_clip.close()
182
- audiotrack = "results/audiotrack.wav"
183
- # client = OpenAI(api_key="sk-proj-KY1qI7zTpsUiJhMUHuNdT3BlbkFJLOjVnTUSpYJi87yUtSEI")
184
- # audiotrack= open("audiotrack.wav", "rb")
185
- transcription = self.client.audio.transcriptions.create(
186
- model="whisper-1",
187
- file=audioclip
188
- )
189
- print(transcription.text)
 
 
 
 
 
 
 
 
 
 
 
 
190
  return transcription.text
191
 
192
  def generate_text(self, inputs, parameters=None):
 
175
  video_name = mp4_file.split("/")[-1]
176
  wav_file = "results/audiotrack.wav"
177
  video_clip = VideoFileClip(mp4_file)
178
+ with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as temp_audio:
179
+ video_clip.audio.write_audiofile(temp_audio.name, codec='pcm_s16le', nbytes=2, fps=16000)
 
180
  video_clip.close()
181
+
182
+ with open(temp_audio.name, 'rb') as audio_file:
183
+ transcription = client.audio.transcriptions.create(
184
+ model="whisper-1",
185
+ file=audio_file
186
+ )
187
+ print(transcription.text)
188
+ os.remove(temp_audio.name)
189
+ # audioclip = video_clip.audio
190
+ # wav_file = audioclip.write_audiofile(wav_file)
191
+ # audioclip.close()
192
+ # video_clip.close()
193
+ # audiotrack = "results/audiotrack.wav"
194
+ # # client = OpenAI(api_key="sk-proj-KY1qI7zTpsUiJhMUHuNdT3BlbkFJLOjVnTUSpYJi87yUtSEI")
195
+ # # audiotrack= open("audiotrack.wav", "rb")
196
+ # transcription = self.client.audio.transcriptions.create(
197
+ # model="whisper-1",
198
+ # file=audioclip
199
+ # )
200
+ # print(transcription.text)
201
  return transcription.text
202
 
203
  def generate_text(self, inputs, parameters=None):