change streamlit sdk version

README.md

@@ -4,7 +4,8 @@ emoji: 🖼️
 colorFrom: blue
 colorTo: purple
 sdk: streamlit
-sdk_version: 1.21.0
+sdk_version: 1.19.0
+python_version: 3.9.13
 app_file: app.py
 pinned: false
 ---

app.py

@@ -8,6 +8,7 @@ from PIL import Image
 from datasets import load_dataset, Dataset, load_from_disk
 from huggingface_hub import login
 import os
+import datasets
 
 
 class GalleryApp:
@@ -25,6 +26,12 @@ class GalleryApp:
                 st.image(image,
                          use_column_width=True,
                 )
+                with st.expander('Similarity Info'):
+                    tab1, tab2 = st.tabs(['Most Similar', 'Least Similar'])
+                    with tab1:
+                        st.image(image, use_column_width=True)
+                    with tab2:
+                        st.image(image, use_column_width=True)
                 for key in info:
                     st.write(f"**{key}**: {items.iloc[idx][key]}")
 
@@ -116,7 +123,8 @@ if __name__ == '__main__':
         print('loading promptBook')
 
         st.session_state.promptBook = pd.DataFrame(load_dataset('NYUSHPRP/ModelCofferMetadata', split='train'))
-        st.session_state.images = load_from_disk(os.path.join(os.getcwd(), 'data', 'promptbook'))
+        st.session_state.images = load_from_disk(os.path.join(os.getcwd(), 'data', 'promptbook'))['train']
+        print(st.session_state.images)
         print('images loaded')
         # st.session_state.promptBook = pd.DataFrame(load_dataset('NYUSHPRP/ModelCofferPromptBook', split='train'))
         st.session_state.promptBook = st.session_state.promptBook.merge(st.session_state.roster[['model_id', 'model_name', 'modelVersion_id', 'modelVersion_name', 'model_download_count']], on=['model_id', 'modelVersion_id'], how='left')

data/download_script.py

@@ -1,8 +1,11 @@
 from datasets import load_dataset, Dataset, load_from_disk
+from diffusers import DPMSolverMultistepScheduler
 
+DPMSolverMultistepScheduler.from_config()
 
 def main():
-    promptbook = load_dataset('NYUSHPRP/ModelCofferPromptBook', split='train')
+    promptbook = load_dataset('NYUSHPRP/ModelCofferPromptBook')
+    print(promptbook)
     promptbook.save_to_disk('./promptbook')
 
     roster = load_dataset('NYUSHPRP/ModelCofferRoster', split='train')

data/promptbook/dataset_dict.json

@@ -0,0 +1 @@
+{"splits": ["train", "validation"]}

data/promptbook/{dataset_info.json → train/dataset_info.json}

@@ -1,50 +1,54 @@
 {
   "citation": "",
-  "dataset_size": 4790396199,
+  "dataset_size": 5164402699,
   "description": "",
   "download_checksums": {
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/8d6cd80dc1021f48d9603ea14a1f74097ed59a53/data/train-00000-of-00010-cd1633256b85c720.parquet": {
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/train-00000-of-00010-2156f01bb59ccb9c.parquet": {
       "num_bytes": 483210906,
       "checksum": null
     },
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/8d6cd80dc1021f48d9603ea14a1f74097ed59a53/data/train-00001-of-00010-5b5a32ac3990b07d.parquet": {
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/train-00001-of-00010-10fc45be38b42231.parquet": {
       "num_bytes": 476302101,
       "checksum": null
     },
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/8d6cd80dc1021f48d9603ea14a1f74097ed59a53/data/train-00002-of-00010-53e82c1a34dc8a24.parquet": {
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/train-00002-of-00010-9e397b3a5cbbdb2e.parquet": {
       "num_bytes": 477247646,
       "checksum": null
     },
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/8d6cd80dc1021f48d9603ea14a1f74097ed59a53/data/train-00003-of-00010-1584b725a142a964.parquet": {
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/train-00003-of-00010-17de07e7e11eb65e.parquet": {
       "num_bytes": 480767762,
       "checksum": null
     },
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/8d6cd80dc1021f48d9603ea14a1f74097ed59a53/data/train-00004-of-00010-71df51519e286b37.parquet": {
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/train-00004-of-00010-3b98d248f078ec45.parquet": {
       "num_bytes": 480021191,
       "checksum": null
     },
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/8d6cd80dc1021f48d9603ea14a1f74097ed59a53/data/train-00005-of-00010-496f4d5e272abbcb.parquet": {
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/train-00005-of-00010-070546d36dfce24e.parquet": {
       "num_bytes": 479774809,
       "checksum": null
     },
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/8d6cd80dc1021f48d9603ea14a1f74097ed59a53/data/train-00006-of-00010-fdeaab3f5af10d27.parquet": {
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/train-00006-of-00010-6992d1b3ab04eacb.parquet": {
       "num_bytes": 476046395,
       "checksum": null
     },
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/8d6cd80dc1021f48d9603ea14a1f74097ed59a53/data/train-00007-of-00010-d2388810ca704795.parquet": {
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/train-00007-of-00010-2aa74b0b7801e933.parquet": {
       "num_bytes": 477758243,
       "checksum": null
     },
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/8d6cd80dc1021f48d9603ea14a1f74097ed59a53/data/train-00008-of-00010-2bc31600a6722b7f.parquet": {
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/train-00008-of-00010-79efbecfb91addf7.parquet": {
       "num_bytes": 479373326,
       "checksum": null
     },
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/8d6cd80dc1021f48d9603ea14a1f74097ed59a53/data/train-00009-of-00010-e3f7e9b1da9d35f7.parquet": {
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/train-00009-of-00010-4abef121572a4382.parquet": {
       "num_bytes": 477778395,
       "checksum": null
+    },
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/validation-00000-of-00001-cdb764fc0db3606c.parquet": {
+      "num_bytes": 373953177,
+      "checksum": null
     }
   },
-  "download_size": 4788280774,
+  "download_size": 5162233951,
   "features": {
     "image": {
       "_type": "Image"
@@ -104,7 +108,7 @@
   },
   "homepage": "",
   "license": "",
-  "size_in_bytes": 9578676973,
+  "size_in_bytes": 10326636650,
   "splits": {
     "train": {
       "name": "train",
@@ -118,6 +122,12 @@
         1586
       ],
       "dataset_name": "parquet"
+    },
+    "validation": {
+      "name": "validation",
+      "num_bytes": 374006500,
+      "num_examples": 560,
+      "dataset_name": "parquet"
     }
   }
 }

data/promptbook/{state.json → train/state.json}

@@ -31,7 +31,7 @@
       "filename": "data-00009-of-00010.arrow"
     }
   ],
-  "_fingerprint": "1c800ac7347ade50",
+  "_fingerprint": "8953d0e6fe589276",
   "_format_columns": null,
   "_format_kwargs": {},
   "_format_type": null,

data/promptbook/validation/data-00000-of-00001.arrow

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:38e8a7f1cfc77e376fbfcdd4988d99d45685b2caaaaf54c402abdf455935957c
+size 374008720

data/promptbook/validation/dataset_info.json

@@ -0,0 +1,133 @@
+{
+  "citation": "",
+  "dataset_size": 5164402699,
+  "description": "",
+  "download_checksums": {
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/train-00000-of-00010-2156f01bb59ccb9c.parquet": {
+      "num_bytes": 483210906,
+      "checksum": null
+    },
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/train-00001-of-00010-10fc45be38b42231.parquet": {
+      "num_bytes": 476302101,
+      "checksum": null
+    },
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/train-00002-of-00010-9e397b3a5cbbdb2e.parquet": {
+      "num_bytes": 477247646,
+      "checksum": null
+    },
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/train-00003-of-00010-17de07e7e11eb65e.parquet": {
+      "num_bytes": 480767762,
+      "checksum": null
+    },
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/train-00004-of-00010-3b98d248f078ec45.parquet": {
+      "num_bytes": 480021191,
+      "checksum": null
+    },
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/train-00005-of-00010-070546d36dfce24e.parquet": {
+      "num_bytes": 479774809,
+      "checksum": null
+    },
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/train-00006-of-00010-6992d1b3ab04eacb.parquet": {
+      "num_bytes": 476046395,
+      "checksum": null
+    },
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/train-00007-of-00010-2aa74b0b7801e933.parquet": {
+      "num_bytes": 477758243,
+      "checksum": null
+    },
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/train-00008-of-00010-79efbecfb91addf7.parquet": {
+      "num_bytes": 479373326,
+      "checksum": null
+    },
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/train-00009-of-00010-4abef121572a4382.parquet": {
+      "num_bytes": 477778395,
+      "checksum": null
+    },
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/9c76b44a3c3319f788ee6fea27d3575618ff3e9c/data/validation-00000-of-00001-cdb764fc0db3606c.parquet": {
+      "num_bytes": 373953177,
+      "checksum": null
+    }
+  },
+  "download_size": 5162233951,
+  "features": {
+    "image": {
+      "_type": "Image"
+    },
+    "image_id": {
+      "dtype": "string",
+      "_type": "Value"
+    },
+    "tag": {
+      "dtype": "string",
+      "_type": "Value"
+    },
+    "model_id": {
+      "dtype": "int64",
+      "_type": "Value"
+    },
+    "modelVersion_id": {
+      "dtype": "int64",
+      "_type": "Value"
+    },
+    "prompt_id": {
+      "dtype": "int64",
+      "_type": "Value"
+    },
+    "size": {
+      "dtype": "string",
+      "_type": "Value"
+    },
+    "seed": {
+      "dtype": "int64",
+      "_type": "Value"
+    },
+    "prompt": {
+      "dtype": "string",
+      "_type": "Value"
+    },
+    "negativePrompt": {
+      "dtype": "string",
+      "_type": "Value"
+    },
+    "cfgScale": {
+      "dtype": "int64",
+      "_type": "Value"
+    },
+    "sampler": {
+      "dtype": "string",
+      "_type": "Value"
+    },
+    "brisque_score": {
+      "dtype": "float64",
+      "_type": "Value"
+    },
+    "clip_score": {
+      "dtype": "float64",
+      "_type": "Value"
+    }
+  },
+  "homepage": "",
+  "license": "",
+  "size_in_bytes": 10326636650,
+  "splits": {
+    "train": {
+      "name": "train",
+      "num_bytes": 4790396199,
+      "num_examples": 7935,
+      "shard_lengths": [
+        1588,
+        1588,
+        1587,
+        1586,
+        1586
+      ],
+      "dataset_name": "parquet"
+    },
+    "validation": {
+      "name": "validation",
+      "num_bytes": 374006500,
+      "num_examples": 560,
+      "dataset_name": "parquet"
+    }
+  }
+}

data/promptbook/validation/state.json

@@ -0,0 +1,13 @@
+{
+  "_data_files": [
+    {
+      "filename": "data-00000-of-00001.arrow"
+    }
+  ],
+  "_fingerprint": "f63d4973dd16d581",
+  "_format_columns": null,
+  "_format_kwargs": {},
+  "_format_type": null,
+  "_output_all_columns": false,
+  "_split": "validation"
+}

data/roster/data-00000-of-00001.arrow

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fe153e2d30eb6f28bcbe03bdbf78d99fee777e03917848575f5dc0715f8c3a64
-size 109016
+oid sha256:4f14f1edb8a119aaee68a138304514ce904d79a9770d9b9f1b2f946538f9d2bd
+size 108712

data/roster/dataset_info.json

@@ -1,14 +1,14 @@
 {
   "citation": "",
-  "dataset_size": 107638,
+  "dataset_size": 107300,
   "description": "",
   "download_checksums": {
-    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferRoster/resolve/ad2d1c28c610862c42861ca20baa814312d06bea/data/train-00000-of-00001-e90719e7788f6218.parquet": {
-      "num_bytes": 16643,
+    "https://huggingface.co/datasets/NYUSHPRP/ModelCofferRoster/resolve/03616d02832904b4d6ac63df0e2aaa780865294a/data/train-00000-of-00001-baa60f73306da841.parquet": {
+      "num_bytes": 16607,
       "checksum": null
     }
   },
-  "download_size": 16643,
+  "download_size": 16607,
   "features": {
     "tag": {
       "dtype": "string",
@@ -45,12 +45,12 @@
   },
   "homepage": "",
   "license": "",
-  "size_in_bytes": 124281,
+  "size_in_bytes": 123907,
   "splits": {
     "train": {
       "name": "train",
-      "num_bytes": 107638,
-      "num_examples": 773,
+      "num_bytes": 107300,
+      "num_examples": 770,
       "dataset_name": "parquet"
     }
   }

data/roster/state.json

@@ -4,7 +4,7 @@
       "filename": "data-00000-of-00001.arrow"
     }
   ],
-  "_fingerprint": "e1e2eff17d4ef545",
+  "_fingerprint": "d1bf8a336edf58a0",
   "_format_columns": null,
   "_format_kwargs": {},
   "_format_type": null,