Update app.py

app.py

@@ -74,8 +74,8 @@ def plot_2d_comparison(dna_sequences_grouped, labels, mapping=mapping_easy):
 
 
 ############################################################# Virus Dataset ########################################################
-ds = load_dataset('Hack90/virus_tiny')
-df = pd.DataFrame(ds['train'])
+#ds = load_dataset('Hack90/virus_tiny')
+df = pd.from_parquet('virus_ds.parquet')
 virus = df['Organism_Name'].unique()
 virus = {v: v for v in virus}
 
@@ -853,8 +853,8 @@ with ui.layout_columns():
 here = Path(__file__).parent
 @render.plot
 def plot():  
-    ds = load_dataset('Hack90/virus_tiny')
-    df = pd.DataFrame(ds['train'])
+    #ds = load_dataset('Hack90/virus_tiny')
+    df = pd.read_parquet('virus_ds.parquet')
     df = df[df['Organism_Name'].isin(input.virus_selector())]
     # group by virus
     grouped = df.groupby('Organism_Name')['Sequence'].apply(list)