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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,7 +7,7 @@ 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):
11
  try:
12
  model = AutoModelForCausalLM.from_pretrained(model_name)
13
  st.write(f"Model {model_name} loaded successfully.")
@@ -26,7 +26,7 @@ with st.sidebar:
26
  💡 Note: No API key required!
27
  ''')
28
  add_vertical_space(5)
29
- st.write('Made with ❤️ by [Data Professor](https://youtube.com/dataprofessor)')
30
 
31
  # Generate empty lists for generated and past.
32
  ## generated stores AI generated responses
@@ -71,5 +71,5 @@ with response_container:
71
  message(st.session_state["generated"][i], key=str(i))
72
 
73
  # Check the model
74
- check_model("gpt3") # For GPT-3
75
  # check_model("gpt4") # For GPT-4
 
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.")
 
26
  💡 Note: No API key required!
27
  ''')
28
  add_vertical_space(5)
29
+ st.write('Made withh Love')
30
 
31
  # Generate empty lists for generated and past.
32
  ## generated stores AI generated responses
 
71
  message(st.session_state["generated"][i], key=str(i))
72
 
73
  # Check the model
74
+ check_model("bert-base-uncased") # For GPT-3
75
  # check_model("gpt4") # For GPT-4