Update main.py

main.py

@@ -17,21 +17,19 @@ from langchain import HuggingFaceHub
 import requests
 
 # 初始化Chatbot
-HUGGINGFACEHUB_API_TOKEN = os.getenv('HUGGINGFACEHUB_API_TOKEN')
+hf_token = os.environ.get('HUGGINGFACEHUB_API_TOKEN')
 repo_id = os.environ.get('repo_id')
-port = os.getenv('port')
+#port = os.getenv('port')
 
 llm = HuggingFaceHub(repo_id=repo_id,
+                     #huggingfacehub_api_token="hf_p***K",
+                     huggingfacehub_api_token=hf_token,
                      model_kwargs={"min_length":1024,
                                    "max_new_tokens":5632, "do_sample":True,
                                    "temperature":0.1,
                                    "top_k":50,
                                    "top_p":0.95, "eos_token_id":49155}) 
 
-#prompt_template = """You are a very helpful AI assistant. Please response to the user's input question with as many details as possible.
-#Question: {user_question}
-#Helpful AI Repsonse:
-#"""
 
 prompt_template = """
 <<SYS>>You are a helpful, respectful and honest assistant. Always answer as helpfully as possible, while being safe.  Your answers should not include any harmful, unethical, racist, sexist, toxic, dangerous, or illegal content. Please ensure that your responses are socially unbiased and positive in nature.