raygiles3 commited on
Commit
cb266ba
1 Parent(s): f955682

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,8 +14,8 @@ if hf_api_token is None:
14
  login(token=hf_api_token, add_to_git_credential=True)
15
 
16
  # Initialize the Whisper processor and model
17
- whisper_processor = WhisperProcessor.from_pretrained("openai/whisper-base")
18
- whisper_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-base")
19
 
20
  # Initialize the summarization model and tokenizer
21
  # Use BART model for summarization
 
14
  login(token=hf_api_token, add_to_git_credential=True)
15
 
16
  # Initialize the Whisper processor and model
17
+ whisper_processor = WhisperProcessor.from_pretrained("openai/whisper-tiny.en")
18
+ whisper_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-tiny.en")
19
 
20
  # Initialize the summarization model and tokenizer
21
  # Use BART model for summarization