raygiles3 commited on
Commit
fdf0af4
·
verified ·
1 Parent(s): d0e774b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -14,14 +14,13 @@ if hf_api_token is None:
14
  # Authenticate with Hugging Face
15
  login(token=hf_api_token, add_to_git_credential=True)
16
 
17
-
18
  # Initialize the Whisper processor and model
19
  whisper_processor = WhisperProcessor.from_pretrained("openai/whisper-base")
20
  whisper_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-base")
21
 
22
  # Initialize the summarization model and tokenizer
23
- # Load LLAMA 7B model with accelerate from local path
24
- model_name = "meta-llama/Llama-2-7b-chat"
25
  with init_empty_weights():
26
  summarization_model = AutoModelForCausalLM.from_pretrained(model_name)
27
 
 
14
  # Authenticate with Hugging Face
15
  login(token=hf_api_token, add_to_git_credential=True)
16
 
 
17
  # Initialize the Whisper processor and model
18
  whisper_processor = WhisperProcessor.from_pretrained("openai/whisper-base")
19
  whisper_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-base")
20
 
21
  # Initialize the summarization model and tokenizer
22
+ # Load LLAMA 7B model with accelerate
23
+ model_name = "meta-llama/Llama-2-7b-hf"
24
  with init_empty_weights():
25
  summarization_model = AutoModelForCausalLM.from_pretrained(model_name)
26