rohangbs commited on
Commit
d35fc36
·
verified ·
1 Parent(s): e482461

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,13 +7,13 @@ tokenizer = GPT2Tokenizer.from_pretrained(model_name)
7
  model = GPT2LMHeadModel.from_pretrained(model_name)
8
 
9
  # Streamlit UI
10
- st.title("Fine-Tuned GPT-2")
11
  st.write("A GPT-2 model fine-tuned for Company dataset.")
12
 
13
  # User input
14
- prompt = st.text_area("Enter your prompt:", height=150)
15
 
16
- if st.button("Generate"):
17
  if prompt.strip():
18
  with st.spinner("Generating..."):
19
  # Generate text
 
7
  model = GPT2LMHeadModel.from_pretrained(model_name)
8
 
9
  # Streamlit UI
10
+ st.title("Chatbot For Company Details")
11
  st.write("A GPT-2 model fine-tuned for Company dataset.")
12
 
13
  # User input
14
+ prompt = st.text_area("Ask your question:", height=150)
15
 
16
+ if st.button("Send"):
17
  if prompt.strip():
18
  with st.spinner("Generating..."):
19
  # Generate text