Update app.py

app.py

@@ -10,10 +10,9 @@ def get_dataset_forward():
     final = []
     dataset = load_dataset("ShoukanLabs/OpenNiji-Dataset", split=f"train[{startimg}:{startimg + 50}]")
     for idx in dataset:
-        print(idx)
-        url = dataset["url"]
-        prompt = dataset["prompt"]
-        style = dataset["style"]
+        url = idx["url"]
+        prompt = idx["prompt"]
+        style = idx["style"]
         final.append((url, f"{prompt}\n\n Style: {style}"))
     startimg += 50
     return final
@@ -24,9 +23,9 @@ def get_dataset_back():
     startimg -= 50
     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"]
+        url = idx["url"]
+        prompt = idx["prompt"]
+        style = idx["style"]
         final.append((url, f"{prompt}\n\n Style: {style}"))
     return final