Update app.py

app.py

@@ -16,14 +16,14 @@ login(token=hf_api_token, add_to_git_credential=True)
 
 
 # Define the local directory where the model will be saved
-local_model_dir = "./LlamaGuard-7b"
+local_model_dir = "meta-llama/Llama-2-7b-chat"
 
 # Create the directory if it doesn't exist
 os.makedirs(local_model_dir, exist_ok=True)
 
 # Download the model and tokenizer
-model = AutoModelForCausalLM.from_pretrained("meta-llama/LlamaGuard-7b", cache_dir=local_model_dir)
-tokenizer = AutoTokenizer.from_pretrained("meta-llama/LlamaGuard-7b", cache_dir=local_model_dir)
+model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-chat", cache_dir=local_model_dir)
+tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-chat", cache_dir=local_model_dir)
 
 # Initialize the Whisper processor and model
 whisper_processor = WhisperProcessor.from_pretrained("openai/whisper-base")