Update app.py

app.py

@@ -12,7 +12,7 @@ ethnicity_labels = ['African-American', 'American_Indian', 'Black', 'Caucasian',
 models = ['DallE', 'SD_14', 'SD_2']
 nos = [1,2,3,4,5,6,7,8,9,10]
 
-ds = load_dataset("color-sorted")
+ds = load_from_disk("color-sorted")
 
 def get_nearest_64(gender, ethnicity, model, no, index):
     df = ds.remove_columns(["image","image_path"]).to_pandas()