Alex commited on
Commit
a6e156d
·
1 Parent(s): b89d00f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,12 +7,12 @@ from transformers import AutoModelForCausalLM
7
 
8
  st.set_page_config(page_title="HugChat - An LLM-powered Streamlit app")
9
 
10
- def check_model(model_name):bert-base-uncased
11
  try:
12
  model = AutoModelForCausalLM.from_pretrained(model_name)
13
  st.write(f"Model {model_name} loaded successfully.")
14
  except Exception as e:
15
- st.write(f"Failed to load model {model_name}. Error: {str(e)}")
16
 
17
  # Sidebar contents
18
  with st.sidebar:
@@ -21,7 +21,7 @@ with st.sidebar:
21
  ## About
22
  This app is an LLM-powered chatbot built using:
23
  - [Streamlit](https://streamlit.io/)
24
- - [OpenAssistant/oasst-sft-6-llama-30b-xor] LLM model
25
 
26
  💡 Note: No API key required!
27
  ''')
 
7
 
8
  st.set_page_config(page_title="HugChat - An LLM-powered Streamlit app")
9
 
10
+ def check_model(model_name):"bert-base-uncased"
11
  try:
12
  model = AutoModelForCausalLM.from_pretrained(model_name)
13
  st.write(f"Model {model_name} loaded successfully.")
14
  except Exception as e:
15
+ st.write(f"Failed to load model {bert-base-uncased}. Error: {str(e)}")
16
 
17
  # Sidebar contents
18
  with st.sidebar:
 
21
  ## About
22
  This app is an LLM-powered chatbot built using:
23
  - [Streamlit](https://streamlit.io/)
24
+ - [bert-base-uncased] LLM model
25
 
26
  💡 Note: No API key required!
27
  ''')