Update main.py

main.py

@@ -15,10 +15,12 @@ import uuid
 import sys
 
 hf_token = os.environ.get('HUGGINGFACEHUB_API_TOKEN')
-repo_id = os.environ.get('repo_id')
+starchat_repo_id = os.environ.get('starchat_repo_id')
+llama2_repo_id=os.environ.get('llama2_repo_id')
 #port = os.getenv('port')
 
-llm = HuggingFaceHub(repo_id=repo_id,
+llm = HuggingFaceHub(#starchat_repo_id=starchat_repo_id,
+                     llama2_repo_id=llama2_repo_id,
                      #huggingfacehub_api_token="hf_p***K",
                      huggingfacehub_api_token=hf_token,
                      model_kwargs={"min_length":100,