Update app.py

app.py

@@ -4,7 +4,8 @@ import json
 import streamlit as st
 
 model_name = "distilbert-base-cased"
-tokenizer = DistilBertTokenizerFast.from_pretrained(model_name)pled by force, as it can have serious consequences. Instead, we should strive for peaceful resolution through dialogue and diplomacy.
+tokenizer = DistilBertTokenizerFast.from_pretrained(model_name)
+model = DistilBertForQuestionAnswering.from_pretrained(model_name)
 
 def format_response(start_index, end_index, raw_answer):
     answer_tokens = tokenizer.convert_tokens_to_string([tokenizer.convert_ids_to_tokens(i)[0] for i in range(start_index, end_index+1)])
@@ -27,10 +28,20 @@ def interactive():
     while query != st.session_state.get('current_query', ''):
         if ('current_query' not in st.session_state) or (st.session_state['current_query'] != query):
             st.session_state['current_query'] = query
-            st.text_input("Ask me something or type 'quit' to exit:", "", key='my_input', on_change=update_displayed_query) # pass the name of the function here
+            st.text_input("Ask me something or type 'quit' to exit:", "", key='my_input')
+            query = st.session_state.get('current_query', '')
+            if query != 'quit':
+                if len(query) > 0:
+                    try:
+                        context = "The capital of France is Paris."
+                        response = get_answers(query, context)
+                        st.write(json.dumps(response))
+                    except Exception as e:
+                        st.write(f"Error occurred: {str(e)}")
 
-        if ('shown_query' not in st.session_state) or (st.session_state['shown_query'] != query):
-            st.session_state['shown_query'] = query
-            if len(query) > 0 and query != 'quit':
-                try:
-                    context
+if __name__ == "__main__":
+    st.set_page_config(layout="wide")
+    st.title('AI Chatbot Built Using Hugging Face and Streamlit')
+    st.subheader('Welcome to Our Intelligent Conversational Agent!')
+    st.write('Please enter your question below or type "quit" to close the session.')
+    interactive()