DarwinAnim8or commited on
Commit
8ac09b0
·
1 Parent(s): 2fc3093

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,9 +1,9 @@
1
  import gradio as gr
2
- from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, WhisperProcessor
3
 
4
  # Load the model and tokenizer
5
  model_id = "openai/whisper-medium"
6
- model = WhisperProcessor.from_pretrained(model_id)
7
  tokenizer = AutoTokenizer.from_pretrained(model_id)
8
 
9
  # Create a WhisperProcessor instance
 
1
  import gradio as gr
2
+ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, WhisperModel, WhisperProcessor
3
 
4
  # Load the model and tokenizer
5
  model_id = "openai/whisper-medium"
6
+ model = WhisperModel.from_pretrained(model_id)
7
  tokenizer = AutoTokenizer.from_pretrained(model_id)
8
 
9
  # Create a WhisperProcessor instance