Update app.py

app.py

@@ -9,11 +9,13 @@ from huggingface_hub import login
 login('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')
 
 # Model name in Hugging Face docs
-model = "meta-llama/Llama-2-7b-chat-hf"
-#'klyang/MentaLLaMA-chat-7B'
+model = 'klyang/MentaLLaMA-chat-7B'
+
+#"meta-llama/Llama-2-7b-chat-hf"
 
-tokenizer = AutoTokenizer.from_pretrained(model, use_auth_token=True)
 
+tokenizer = AutoTokenizer.from_pretrained(model, use_auth_token=True)
+#llama_pipeline = AutoModelForCausalLM.from_pretrained(model)
 
 llama_pipeline = pipeline(
     "text-generation",  # LLM task