Imageye commited on
Commit
707c124
·
verified ·
1 Parent(s): 12f2a79

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -4,12 +4,13 @@ from youtube_transcript_api import YouTubeTranscriptApi
4
  import re
5
  import tempfile
6
  import os
7
- from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
8
  import torch
 
9
 
10
  # Load the Whisper model and processor from Hugging Face
11
- processor = AutoProcessor.from_pretrained("openai/whisper-large-v3")
12
- model = AutoModelForSpeechSeq2Seq.from_pretrained("openai/whisper-large-v3")
13
 
14
  def transcribe_audio(file_path):
15
  # Load audio file
 
4
  import re
5
  import tempfile
6
  import os
7
+ from transformers import WhisperProcessor, WhisperForConditionalGeneration
8
  import torch
9
+ import librosa
10
 
11
  # Load the Whisper model and processor from Hugging Face
12
+ processor = WhisperProcessor.from_pretrained("openai/whisper-large-v2")
13
+ model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-large-v2")
14
 
15
  def transcribe_audio(file_path):
16
  # Load audio file