Update main.py

main.py

@@ -437,7 +437,9 @@ async def start():
     #    temperature=1.0,
     #    streaming=True
     #)
+    from huggingface_hub import login
     from langchain_huggingface.llms import HuggingFacePipeline
+    login(token=os.environ['HUGGINGFACEHUB_API_TOKEN'])
     model = HuggingFacePipeline.from_model_id(
         model_id="mistralai/Mistral-Small-Instruct-2409",
         task="text-generation",