selectbox bug fix

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from datasets import load_dataset
+import pandas as pd
+
 
 # finetuned model
 language_model_path = "juliaannjose/finetuned_model"
@@ -19,12 +21,15 @@ with st.spinner("Setting up the app..."):
         val_filing_start_date="2016-01-22",
         val_filing_end_date="2016-01-31",
     )
+    df_train = pd.DataFrame(dataset_dict["train"])
+    df_val = pd.DataFrame(dataset_dict["validation"])
+    df = pd.concat([df_train, df_val], ignore_index=True)
 
 
 # drop down menu with patent numbers
 _patent_id = st.selectbox(
     "Select the Patent Number",
-    options=dataset_dict["train"]["patent_number"],
+    options=df["patent_number"],
 )
 
 
@@ -32,8 +37,8 @@ _patent_id = st.selectbox(
 def get_abs_claim(_pid):
     # get abstract and claim corresponding to this patent id
     st.write(_pid)
-    _abs = dataset_dict["train"][["patent_number"] == _pid]["abstract"]
-    _cl = dataset_dict["train"][["patent_number"] == _pid]["claims"]
+    _abs = df.loc[["patent_number"] == _pid]["abstract"]
+    _cl = df.loc[["patent_number"] == _pid]["claims"]
     st.write(_abs)
     return _abs, _cl