baharbhz commited on
Commit
601054b
·
verified ·
1 Parent(s): 9b13622

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -4,7 +4,10 @@ import torchaudio
4
  import torch
5
  import librosa
6
  import numpy as np
7
- import moviepy.editor as mp
 
 
 
8
  from transformers import Wav2Vec2Processor, Wav2Vec2ForCTC
9
 
10
  model_name = "m3hrdadfi/wav2vec2-large-xlsr-persian"
@@ -30,7 +33,8 @@ def speech_to_text(audio_path):
30
 
31
 
32
  def video_to_text(video_path):
33
- video = mp.VideoFileClip(video_path)
 
34
  audio_path = "extracted_audio.wav"
35
  video.audio.write_audiofile(audio_path, codec="pcm_s16le")
36
  return speech_to_text(audio_path)
 
4
  import torch
5
  import librosa
6
  import numpy as np
7
+ # import moviepy.editor as mp
8
+ import moviepy
9
+ from moviepy.video.io.VideoFileClip import VideoFileClip
10
+
11
  from transformers import Wav2Vec2Processor, Wav2Vec2ForCTC
12
 
13
  model_name = "m3hrdadfi/wav2vec2-large-xlsr-persian"
 
33
 
34
 
35
  def video_to_text(video_path):
36
+ # video = mp.VideoFileClip(video_path)
37
+ video = VideoFileClip(video_path)
38
  audio_path = "extracted_audio.wav"
39
  video.audio.write_audiofile(audio_path, codec="pcm_s16le")
40
  return speech_to_text(audio_path)