Update app.py

app.py

@@ -23,7 +23,7 @@ mpl.rcParams.update(mpl.rcParamsDefault)
 df = pd.read_parquet('virus_ds.parquet')
 virus = df['Organism_Name'].unique()
 virus = {v: v for v in virus}
-df_new = pd.read_parquet("distro.parquet", columns= ['organism_name']).tolist()
+df_new = pd.read_parquet("distro.parquet", columns= ['organism_name'])['organism_name'].tolist()
 MASTER_DF = pd.read_parquet("distro.parquet")
 virus_new = {v: v for v in df_new}
 loss_typesss = pd.read_csv("training_data_5.csv")['loss_type'].unique().tolist()