Spaces:

nguyennghia0902
/

proj02_textmining

Sleeping

nguyennghia0902 commited on May 29

Commit

d6d8560

•

1 Parent(s): 042b4ce

Update SampleQA.py

Files changed (1) hide show

SampleQA.py CHANGED Viewed

@@ -57,7 +57,10 @@ def main():
     if st.button("Sample & Predict"):
         prediction = ""
         prediction = predict(question, text)
-        st.success(prediction)
 if __name__ == "__main__":
     main()

     if st.button("Sample & Predict"):
         prediction = ""
         prediction = predict(question, text)
+        if prediction == "":
+            st.error(prediction)
+        else:
+            st.success(prediction)
 if __name__ == "__main__":
     main()