Update main.py

main.py

@@ -11,11 +11,10 @@ import sys
 
 load_dotenv()
 hf_token = os.environ.get('HUGGINGFACEHUB_API_TOKEN')
-#starchat_repo_id = os.environ.get('starchat_repo_id')
-#repo_id=os.environ.get('repo_id')
+repo_id=os.environ.get('repo_id')
 #port = os.getenv('port')
 
-llm = HuggingFaceHub(repo_id=starchat_repo_id,  # for StarChat
+llm = HuggingFaceHub(repo_id=repo_id,  # for StarChat
                      huggingfacehub_api_token=hf_token,
                      model_kwargs={"min_length": 512,  # for StarChat
                                    "max_new_tokens": 1024, "do_sample": True,  # for StarChat
@@ -44,6 +43,8 @@ async def chat(request: Request):
             print(user_query)
             docs = loader.load()
             result = chain.run(docs)
+            
+            
             print("AI Summarization: " + result)
             return {'response': result}
         except Exception as e: