Spaces:
Runtime error
Runtime error
Alex commited on
Commit ·
b89d00f
1
Parent(s): 311a01f
Update app.py
Browse files
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
|
| 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("
|
| 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
|