Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+from transformers import AutoTokenizer, AutoModelForQuestionAnswering, pipeline
 
 st.title('Question-Answering NLU')
 
@@ -42,4 +43,31 @@ question-answering data that can be used by QANLU. MATIS++ includes
 the original English version of ATIS and a translation into eight languages: German, Spanish, French, 
 Japanese, Hindi, Portuguese, Turkish, and Chinese. 
 
-''')
+''')
+
+elif menu == "Evaluation":
+    st.header('QANLU Evaluation')
+    tokenizer = AutoTokenizer.from_pretrained("AmazonScience/qanlu", use_auth_token=True)
+
+    model = AutoModelForQuestionAnswering.from_pretrained("AmazonScience/qanlu", use_auth_token=True)
+
+    qa_pipeline = pipeline('question-answering', model=model, tokenizer=tokenizer)
+    
+    context = st.text_input(
+        'Please enter the context:',
+        value="I want a cheap flight to Boston."
+    )
+    question = st.text_input(
+        'Please enter the question:',
+        value="What is the destination?"
+    )
+
+
+    qa_input = {
+      'context': 'Yes. No. ' + context,
+      'question': question
+    }
+
+    if st.button('Ask QANLU'):
+        answer = qa_pipeline(qa_input)
+        st.write(answer)