Update app.py

app.py

@@ -10,6 +10,8 @@ from langchain.chains.combine_documents import create_stuff_documents_chain
 from langchain_community.embeddings import HuggingFaceBgeEmbeddings
 from langchain_community.llms import CTransformers
 from ctransformers import AutoModelForCausalLM
+from langchain.llms import HuggingFaceHub
+from transformers import AutoTokenizer
 # from dotenv import load_dotenv
 
 # load_dotenv()
@@ -82,7 +84,7 @@ def get_response(user_input):
     #                 lib="avx2", # for CPU
     # )
 
-    llm_model == "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
+    llm_model = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
     llm = HuggingFaceHub(
         repo_id=llm_model, 
         model_kwargs={"temperature": temperature, "max_new_tokens": 250, "top_k": top_k}