Update main.py

main.py

@@ -58,13 +58,14 @@ llm = HuggingFaceHub(repo_id=repo_id,
                                    "top_k":50,
                                    "top_p":0.95, "eos_token_id":49155})
 
-prompt_template = """
-You are a very helpful AI assistant. Please ONLY use {context} to answer the user's question {question}. If you don't know the answer, just say that you don't know. DON'T try to make up an answer.
-Your response should be full and easy to understand.
-"""
-PROMPT = PromptTemplate(template=prompt_template, input_variables=["context", "question"])
-
-chain = load_qa_chain(llm=llm, chain_type="stuff", prompt=PROMPT)
+#prompt_template = """
+#You are a very helpful AI assistant. Please ONLY use {context} to answer the user's question {question}. If you don't know the answer, just say that you don't know. DON'T try to make up an answer.
+#Your response should be full and easy to understand.
+#"""
+#PROMPT = PromptTemplate(template=prompt_template, input_variables=["context", "question"])
+
+#chain = load_qa_chain(llm=llm, chain_type="stuff", prompt=PROMPT)
+chain = load_qa_chain(llm=llm, chain_type="stuff")
 
 app = FastAPI()