Update app.py

app.py

@@ -8,7 +8,7 @@ startimg = 0
 def get_dataset_forward():
     global startimg
     final = []
-    dataset = load_dataset("ShoukanLabs/OpenNiji-Dataset", split=f"train[{startimg}:{startimg + 50}]")
+    dataset = load_dataset("ShoukanLabs/OpenNiji-Dataset-Aesthetic-Finetune", split=f"train[{startimg}:{startimg + 50}]")
     for idx in dataset:
         url = idx["url"]
         prompt = idx["prompt"]
@@ -26,7 +26,8 @@ def get_dataset_back():
         url = idx["url"]
         prompt = idx["prompt"]
         style = idx["style"]
-        final.append((url, f"{prompt}\n\n Style: {style}"))
+        score = idx["score"]
+        final.append((url, f"{prompt}\n\n Style: {style}\n\n Aesthetic: {score}"))
     return final
 
 with gr.Blocks() as demo: