latest changes

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import wandb
 from transformers import pipeline
 from transformers import AutoTokenizer, AutoModelForTokenClassification
+import pandas as pd
 
 x = st.slider('Select a value')
 st.write(x, 'squared is', x * x)
@@ -21,10 +22,26 @@ def load_trained_model():
     ner_pipeline = pipeline("ner", model=model, tokenizer = tokenizer)
     return ner_pipeline
 
-def load_data():
+def load_random_examples(dataset_name, num_examples=5):
+    """
+    Load random examples from the specified Hugging Face dataset.
+    Args:
+        dataset_name (str): The name of the dataset to load.
+        num_examples (int): The number of random examples to load.
+    Returns:
+        pd.DataFrame: A DataFrame containing the random examples.
+    """
+    # Load the dataset
     from datasets import load_dataset
-    dat_CW = load_dataset("surrey-nlp/PLOD-CW")
-
+    dataset = load_dataset("surrey-nlp/PLOD-CW")
+    
+    # Convert the dataset to a pandas DataFrame
+    df = pd.DataFrame(dataset['train'])
+    
+    # Select random examples
+    random_examples = df.sample(n=num_examples)
+    
+    return random_examples
 
 def render_entities(tokens, entities):
     """
@@ -66,6 +83,57 @@ def render_entities(tokens, entities):
     table_data = {"Token": tokens, "Entity": entities}
     st.table(table_data)
 
+def render_random_examples():
+    """
+    Render random examples from the PLOD-CW dataset in a Streamlit table.
+    """
+    # Load random examples
+    random_examples = load_random_examples("surrey-nlp/PLOD-CW")
+    
+    
+    # Custom CSS for chilled and cool theme
+    st.markdown("""
+        <style>
+        body {
+            font-family: 'Arial', sans-serif;
+            background-color: #f0f0f5;
+            color: #333333;
+        }
+        table {
+            width: 100%;
+            border-collapse: collapse;
+        }
+        th, td {
+            padding: 12px;
+            text-align: left;
+            border-bottom: 1px solid #dddddd;
+        }
+        th {
+            background-color: #4CAF50;
+            color: white;
+        }
+        tr:hover {
+            background-color: #f5f5f5;
+        }
+        </style>
+        """, unsafe_allow_html=True)
+
+    # Title and description
+    st.title("Random Examples from PLOD-CW")
+    st.write("This table shows 5 random examples from the PLOD-CW dataset in a cool and chilled theme.")
+
+    # Add a button to select a different set of random samples
+    if st.button('Show another set of random examples'):
+        st.session_state['random_examples'] = load_random_examples("surrey-nlp/PLOD-CW")
+
+    # Load random examples if not already loaded
+    if 'random_examples' not in st.session_state:
+        st.session_state['random_examples'] = load_random_examples("surrey-nlp/PLOD-CW")
+
+    # Display the table
+    st.table(st.session_state['random_examples'])
+
+    
 def prep_page():
     model = load_trained_model()
 
@@ -120,6 +188,10 @@ def prep_page():
 
         render_entities(text, entities)
 
+        render_random_examples()
+
+
+
 if __name__ == '__main__':
     
     prep_page()