tushar-r-pawar commited on
Commit
5742ae6
·
verified ·
1 Parent(s): e8254f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -13,18 +13,18 @@ api_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
13
 
14
  # Initialize model and tokenizer using the AutoModel from AirLLM
15
  MAX_LENGTH = 128
16
- model = AutoModel.from_pretrained("garage-bAInd/Platypus2-70B-instruct")
17
 
18
  # Streamlit app configuration
19
  st.set_page_config(
20
- page_title="Conversational Chatbot with Platypus2-70B and AirLLM",
21
  page_icon="🤖",
22
  layout="wide",
23
  initial_sidebar_state="expanded",
24
  )
25
 
26
  # App title
27
- st.title("Conversational Chatbot with Platypus2-70B and AirLLM")
28
 
29
  # Sidebar configuration
30
  st.sidebar.header("Chatbot Configuration")
@@ -91,6 +91,6 @@ if st.button("Send"):
91
  st.sidebar.markdown(
92
  """
93
  ### About
94
- This is a conversational chatbot built using the Platypus2-70B model and AirLLM.
95
  """
96
  )
 
13
 
14
  # Initialize model and tokenizer using the AutoModel from AirLLM
15
  MAX_LENGTH = 128
16
+ model = AutoModel.from_pretrained("internlm/internlm2_5-7b-chat-gguf")
17
 
18
  # Streamlit app configuration
19
  st.set_page_config(
20
+ page_title="Conversational Chatbot with internlm2_5-7b-chat and AirLLM",
21
  page_icon="🤖",
22
  layout="wide",
23
  initial_sidebar_state="expanded",
24
  )
25
 
26
  # App title
27
+ st.title("Conversational Chatbot with internlm2_5-7b-chat and AirLLM")
28
 
29
  # Sidebar configuration
30
  st.sidebar.header("Chatbot Configuration")
 
91
  st.sidebar.markdown(
92
  """
93
  ### About
94
+ This is a conversational chatbot built using the internlm2_5-7b-chat model and AirLLM.
95
  """
96
  )