raygiles3 commited on
Commit
14ee4cf
·
verified ·
1 Parent(s): d9b9f7e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ whisper_model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-
20
 
21
  # Initialize the summarization model and tokenizer
22
  # Load LLAMA 7B model with accelerate
23
- model_name = "models/meta-llama/Llama-2-7b-hf"
24
  with init_empty_weights():
25
  summarization_model = AutoModelForCausalLM.from_pretrained(model_name)
26
 
 
20
 
21
  # Initialize the summarization model and tokenizer
22
  # Load LLAMA 7B model with accelerate
23
+ model_name = "/Llama-2-7b-hf"
24
  with init_empty_weights():
25
  summarization_model = AutoModelForCausalLM.from_pretrained(model_name)
26