Update app.py

app.py

@@ -9,10 +9,10 @@ def get_dataset_forward():
     global startimg
     final = []
     dataset = load_dataset("ShoukanLabs/OpenNiji-Dataset", split=f"train[{startimg}:{startimg + 50}]")
-    for idx in dataset
-        url = dataset["url"][idx]
-        prompt = dataset["prompt"][idx]
-        style = dataset["style"][idx]
+    for idx in dataset:
+        url = dataset["url"]
+        prompt = dataset["prompt"]
+        style = dataset["style"]
         final.append((url, f"{prompt}\n\n Style: {style}"))
     startimg += 50
     return final
@@ -21,10 +21,11 @@ def get_dataset_back():
     global startimg
     final = []
     startimg -= 50
-    for idx in range(startimg, startimg + 50):
-        url = dataset["train"]["url"][idx]
-        prompt = dataset["train"]["prompt"][idx]
-        style = dataset["train"]["style"][idx]
+    dataset = load_dataset("ShoukanLabs/OpenNiji-Dataset", split=f"train[{startimg}:{startimg + 50}]")
+    for idx in dataset:
+        url = dataset["url"]
+        prompt = dataset["prompt"]
+        style = dataset["style"]
         final.append((url, f"{prompt}\n\n Style: {style}"))
     return final