Update app.py

app.py

@@ -28,14 +28,17 @@ _patent_id = st.selectbox(
     dataset_dict["train"]["patent_number"],
 )
 
-if _patent_id:
+# display abstract and claim
+@st.cache(persist=True)
+def get_abs_claim(_patent_id):
     # get abstract and claim corresponding to this patent id
     _abstract = dataset_dict["train"][["patent_number"] == _patent_id]["abstract"]
     _claim = dataset_dict["train"][["patent_number"] == _patent_id]["claims"]
+    return _abstract,_claim
 
-    # display abstract and claim
-    st.write(_abstract)
-    st.write(_claim)
+_abstract,_claim = get_abs_claim(_patent_id)
+st.write(_abstract)
+st.write(_claim)
 
 # when submit button clicked, run the model and get result
 if st.button("Submit"):