Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
+st.title("custom chatbot with HF model")
+
 model_name = "facebook/blenderbot-400M-distill"
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
@@ -19,5 +21,7 @@ def get_response(input_text):
 
     return response
 
-answer = get_response("is apple a fruit?")
-st.write(answer)
+user_input = st.text_input("Enter your query here..")
+if st.button("Get Response") and user_input:
+    answer = get_response(user_input)
+    st.write(answer)