Update app.py

app.py

@@ -1,10 +1,14 @@
 import streamlit as st
 from transformers import pipeline
 
-question_answerer = pipeline("question-answering")
-text= st.text_area('enter your context')
-ques = st.text_input('enter your question')
+st.title('QA ')
+model_checkpoint = 'Diezu/dieumrc'
+question_answerer = pipeline("question-answering", model=model_checkpoint)
 
-if st.button('ANSWER'):
-    out=question_answerer(ques,text)
-    st.json(out)
+context = st.text_area('CONTEXT')
+question = st.text_input('QUESTION')
+
+st.title('Predict Flight Price ')
+
+    if st.button('ANSWER'):
+        print(question_answerer(question,context))