Update app.py

app.py

@@ -10,7 +10,7 @@ language_model_path = "juliaannjose/finetuned_model"
 
 # load the dataset to
 # use the patent number, abstract and claim columns for UI
-with st.spinner("Setting up the app..."):
+with st.spinner("Loading..."):
     dataset_dict = load_dataset(
         "HUPD/hupd",
         name="sample",
@@ -36,17 +36,16 @@ _patent_id = st.selectbox(
 # display abstract and claim
 def get_abs_claim(_pid):
     # get abstract and claim corresponding to this patent id
-    st.write(_pid)
-    _abs = df.loc["patent_number" == _pid]["abstract"]
-    _cl = df.loc["patent_number" == _pid]["claims"]
-    st.write(_abs)
-    return _abs, _cl
+    _abs = df.loc[df["patent_number"] == _pid]["abstract"]
+    _cl = df.loc[df["patent_number"] == _pid]["claims"]
+    return _abs.values[0], _cl.values[0]
 
 
-st.write(_patent_id)
 _abstract, _claim = get_abs_claim(_patent_id)
-# st.write(_abstract)  # display abstract
-# st.write(_claim)  # display claims
+st.title("Abstract:")  # display abstract
+st.write(_abstract)
+st.title("Claim:")  # display claims
+st.write(_claim)
 
 
 # model and tokenizer initialization
@@ -77,4 +76,5 @@ if st.button("Submit"):
 
     predicted_class_id = logits.argmax().item()
     pred_label = id2label[predicted_class_id]
+    st.title("Predicted Patentability")
     st.write(pred_label)