Update app.py

app.py

@@ -1,11 +1,13 @@
 import streamlit as st
 from langchain.llms import CTransformers
 from langchain.prompts import PromptTemplate
+from transformers import AutoModelForCausalLM
 
+model = AutoModelForCausalLM.from_pretrained("karthiksagarn/Blog-Generation-App")
 
 ## Function to get response form the LLama2 Model
 def getLLamaResponse(input_text, no_of_words, blog_style):
-    llm = CTransformers(model='karthiksagarn/Blog-Generation-App',
+    llm = CTransformers(model=model,
                         model_type='llama',
                         config={"max_new_tokens": 256,
                                 'temperature': 0.01})