Update app.py
Browse files
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("
|
| 11 |
st.write("A GPT-2 model fine-tuned for Company dataset.")
|
| 12 |
|
| 13 |
# User input
|
| 14 |
-
prompt = st.text_area("
|
| 15 |
|
| 16 |
-
if st.button("
|
| 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
|