IamVicky111 commited on
Commit
1f88095
·
verified ·
1 Parent(s): 94c3bc5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
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.")