Update main.py

main.py

@@ -46,11 +46,12 @@ logger = logging.getLogger(__name__)
 # -------------------------
 # Logs into Hugging Face or uses a fallback to get cookies.
 try:
-    if os.getenv("HF_EMAIL") and os.getenv("HF_PASSWORD"):
-        from hugchat.login import Login
+    if HF_EMAIL and HF_PASSWORD:
         logger.info("Logging into Hugging Face using provided credentials.")
+       # cookie_path_dir = "./cookies/" # NOTE: trailing slash (/) is required to avoid errors
         sign = Login(HF_EMAIL, HF_PASSWORD)
-        cookies = sign.login()
+        cookies = sign.login(save_cookies=True)
+        print("cookies: ", cookies)
     else:
         logger.info("No HF_EMAIL/HF_PASSWORD provided. Using fallback method to get cookies.")
         cookies = requests.get("https://huggingface.co/chat/").cookies