Update main.py

main.py

@@ -4,6 +4,7 @@ from langchain_community.vectorstores import FAISS
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.llms import HuggingFacePipeline
 from transformers import pipeline
+from langchain import PromptTemplate
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 from langchain.chains import RetrievalQA
 import torch