Update main.py

main.py

@@ -19,8 +19,8 @@ starchat_repo_id = os.environ.get('starchat_repo_id')
 llama2_repo_id=os.environ.get('llama2_repo_id')
 #port = os.getenv('port')
 
-llm = HuggingFaceHub(llama2_repo_id=llama2_repo_id,
-                     #starchat_repo_id=starchat_repo_id,                     
+llm = HuggingFaceHub(repo_id=llama2_repo_id,  #for Llama2
+                     #repo_id=starchat_repo_id,     #for StarChat                  
                      huggingfacehub_api_token=hf_token,
                      model_kwargs={#"min_length":512,  #for StarChat
                                    "min_length":1024,  #for Llama2