Update main.py
Browse files
main.py
CHANGED
@@ -47,18 +47,15 @@ logger = logging.getLogger(__name__)
|
|
47 |
# -------------------------
|
48 |
# Logs into Hugging Face or uses a fallback to get cookies.
|
49 |
try:
|
50 |
-
|
51 |
-
|
52 |
-
|
53 |
-
|
54 |
-
print("cookies: ", cookies)
|
55 |
-
else:
|
56 |
-
logger.info("No HF_EMAIL/HF_PASSWORD provided. Using fallback method to get cookies.")
|
57 |
-
cookies = requests.get("https://huggingface.co/chat/").cookies
|
58 |
except Exception as e:
|
59 |
-
|
|
|
|
|
60 |
|
61 |
-
|
62 |
# Create a ChatBot instance from the hugchat library.
|
63 |
chatbot = hugchat.ChatBot(cookies=cookies.get_dict())
|
64 |
|
|
|
47 |
# -------------------------
|
48 |
# Logs into Hugging Face or uses a fallback to get cookies.
|
49 |
try:
|
50 |
+
logger.info("Logging into Hugging Face using provided credentials.")
|
51 |
+
sign = Login(HF_EMAIL, HF_PASSWORD)
|
52 |
+
cookies = sign.login()
|
53 |
+
print("cookies: ", cookies)
|
|
|
|
|
|
|
|
|
54 |
except Exception as e:
|
55 |
+
logger.error(f"Error in Logging-into Hugging Chat: {e}")
|
56 |
+
logger.info("Using fallback method to get cookies.")
|
57 |
+
cookies = requests.get("https://huggingface.co/chat/").cookies
|
58 |
|
|
|
59 |
# Create a ChatBot instance from the hugchat library.
|
60 |
chatbot = hugchat.ChatBot(cookies=cookies.get_dict())
|
61 |
|