Spaces:

DrishtiSharma
/

chat-with-patent-dataset

Running

DrishtiSharma commited on Jan 26

Commit

085efcc

verified ·

1 Parent(s): 5f64a77

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -55,7 +55,7 @@ def load_dataset_into_session():
         )
         if st.button("Load Hugging Face Dataset"):
             try:
-                dataset = load_dataset(dataset_name, name="all", split="train", trust_remote_code=True, uniform_split=True)
                 if hasattr(dataset, "to_pandas"):
                     st.session_state.df = dataset.to_pandas()
                 else:

         )
         if st.button("Load Hugging Face Dataset"):
             try:
+                dataset = load_dataset(dataset_name, name="sample", split="train", trust_remote_code=True, uniform_split=True)
                 if hasattr(dataset, "to_pandas"):
                     st.session_state.df = dataset.to_pandas()
                 else: