Update app.py

app.py

@@ -2,10 +2,14 @@ import streamlit as st
 from transformers import pipeline
 
 # Function to get response from LLaMA 2 model
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
+model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-chat-hf")
 
 def getLLamaresponse(input_text, keywords, blog_style):
     # Load the LLaMA 2 model from Hugging Face
-    model_name = "openai-community/gpt2"
+    model_name = model
     llm = pipeline('text-generation', model=model_name)
 
     # Prompt Template