raygiles3 commited on
Commit
4a65e74
·
verified ·
1 Parent(s): f5f8752

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
  # Initialize the summarization model and tokenizer
21
  # Use a smaller version of the Llama model and load in FP16
22
  summarization_model = AutoModelForCausalLM.from_pretrained(
23
- "meta-llama/Llama-2-3b-hf",
24
  torch_dtype=torch.float16,
25
  low_cpu_mem_usage=True
26
  )
 
20
  # Initialize the summarization model and tokenizer
21
  # Use a smaller version of the Llama model and load in FP16
22
  summarization_model = AutoModelForCausalLM.from_pretrained(
23
+ "meta-llama/LlamaGuard-7b",
24
  torch_dtype=torch.float16,
25
  low_cpu_mem_usage=True
26
  )