Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,6 +2,11 @@ import torch
|
|
| 2 |
import streamlit as st
|
| 3 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
| 4 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
# Choose your desired free model from the Hugging Face Hub
|
| 6 |
model_name = "t5-small" # Replace with your choice (e.g., facebook/bart-base or EleutherAI/gpt-neo-125M)
|
| 7 |
|
|
@@ -14,7 +19,6 @@ st.set_page_config(page_title="LangChain Demo", page_icon=":robot:")
|
|
| 14 |
st.header("Hey, I'm your Chat GPT")
|
| 15 |
|
| 16 |
|
| 17 |
-
|
| 18 |
if "sessionMessages" not in st.session_state:
|
| 19 |
st.session_state.sessionMessages = [
|
| 20 |
SystemMessage(content="You are a helpful assistant.")
|
|
|
|
| 2 |
import streamlit as st
|
| 3 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
| 4 |
|
| 5 |
+
system_message = "Welcome to my Chatbot! Ask me anything."
|
| 6 |
+
|
| 7 |
+
# Display the message using Streamlit's `write` function
|
| 8 |
+
st.write(system_message)
|
| 9 |
+
|
| 10 |
# Choose your desired free model from the Hugging Face Hub
|
| 11 |
model_name = "t5-small" # Replace with your choice (e.g., facebook/bart-base or EleutherAI/gpt-neo-125M)
|
| 12 |
|
|
|
|
| 19 |
st.header("Hey, I'm your Chat GPT")
|
| 20 |
|
| 21 |
|
|
|
|
| 22 |
if "sessionMessages" not in st.session_state:
|
| 23 |
st.session_state.sessionMessages = [
|
| 24 |
SystemMessage(content="You are a helpful assistant.")
|