Update QuestionAnswering.py

QuestionAnswering.py

@@ -3,7 +3,7 @@ import streamlit as st
 import tensorflow as tf
 from transformers import ElectraTokenizerFast, TFElectraForQuestionAnswering
 
-model_hf = "nguyennghia0902/bestfailed_electra-small-discriminator_5e-05_16"
+model_hf = "nguyennghia0902/electra-small-discriminator_0.0005_32"
 tokenizer = ElectraTokenizerFast.from_pretrained(model_hf)
 reload_model = TFElectraForQuestionAnswering.from_pretrained(model_hf)
 
@@ -32,12 +32,12 @@ def main():
     question = st.text_area(
         "QUESTION: Please enter a question:",
         placeholder="Enter your question here",
-        height=15,
+        height=5,
     )
     text = st.text_area(
         "CONTEXT: Please enter a context:",
         placeholder="Enter your context here",
-        height=100,
+        height=200,
     )
 
     prediction = ""
@@ -69,7 +69,10 @@ def main():
             return
 
         prediction = predict(stripped_question, stripped_text)
-        st.success(prediction)
+        if prediction == "":
+            st.error(prediction)
+        else:
+            st.success(prediction)
 
 if __name__ == "__main__":
     main()