CosmoS / app.py
CosmoAI's picture
Update app.py
5eb08a7
raw
history blame
991 Bytes
from hugchat import hugchat
from hugchat.login import Login
import streamlit as st
# Log in to huggingface and grant authorization to huggingchat
sign = Login([email protected], @_3SegZXGjrjff/)
cookies = sign.login()
# Save cookies to the local directory
cookie_path_dir = "./cookies_snapshot"
sign.saveCookiesToDir(cookie_path_dir)
# Create a ChatBot
chatbot = hugchat.ChatBot(cookies=cookies.get_dict()) # or cookie_path="usercookies/<email>.json"
print(chatbot.chat("Hi!"))
# Create a new conversation
id = chatbot.new_conversation()
chatbot.change_conversation(id)
# Get conversation list
conversation_list = chatbot.get_conversation_list()
if uinput := st.chat_input("How can I help you?"):
with st.chat_message:
st.markdown(chatbot.chat(uinput))
# # Switch model (default: meta-llama/Llama-2-70b-chat-hf. )
# chatbot.switch_llm(0) # Switch to `OpenAssistant/oasst-sft-6-llama-30b-xor`
# chatbot.switch_llm(1) # Switch to `meta-llama/Llama-2-70b-chat-hf`