Update SimpleChatbot.py

SimpleChatbot.py

@@ -1,10 +1,8 @@
 # Q&A Chatbot
+import langchain_community
 from langchain_community.llms import HuggingFaceEndpoint
 from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
-
-# from langchain.llms import OpenAI
-# uncomment above line when I have credit in OpenAI
 from dotenv import load_dotenv
 load_dotenv() # take environment variables from .env
 
@@ -15,10 +13,7 @@ import os
 
 def get_model_response(question):
     llm = HuggingFaceEndpoint(
-        repo_id="mistralai/Mistral-7B-Instruct-v0.2", max_length=128, temperature=0.5, token=os.getenv("HUGGINGFACEHUB_API_TOKEN"))
-    
-    # llm = OpenAI(openai_api_key=os.getenv("OPENAI_API_KEY"), model_name="text-davinci-003", temperature=0.6, max_tokens=64)
-    # uncomment above line and use it instead of HuggingfaceEndpoint when I have credit in OpenAI.
+        repo_id="mistralai/Mistral-7B-Instruct-v0.2", max_length=128, temperature=0.5)
     template = """Question: {question}
     Answer:"""
     prompt = PromptTemplate.from_template(template)