Untitled1.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "10676e1f-9a7d-453f-9334-246ebb2142c9",
   "metadata": {},
   "outputs": [],
   "source": [
    "datapath = 'readsy/stories/'\n",
    "pairpath = 'readsy/pairs/readsy_story_pairs0407.csv'\n",
    "from transformers import AutoTokenizer\n",
    "tokenizer = AutoTokenizer.from_pretrained(\"unsloth/gemma-2b-bnb-4bit\")\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "7d73d0b5-7356-4c06-96bc-91dc807bcb0d",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "the total number of pairs is  100\n",
      "the number of effective pairs is  91\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "No chat template is set for this tokenizer, falling back to a default class-level template. This is very error-prone, because models are often trained with templates different from the class default! Default chat templates are a legacy feature and will be removed in Transformers v4.43, at which point any code depending on them will stop working. We recommend setting a valid chat template before then to ensure that this model continues working without issues.\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Index(['prompt_id', 'prompt', 'story_id', 'story_title', 'story_author',\n",
      "       'story_url', 'link', 'genre', 'is_sensitive', 'categories', 'likes',\n",
      "       'story_text', 'posted_date', 'comments'],\n",
      "      dtype='object')\n",
      "the columns of train is  Index(['prompt_id', 'story1_id', 'story2_id', 'time_lag', 'least_likes'], dtype='object')\n",
      "prompt_1234\n",
      "78559    Write a story about someone who finds somethin...\n",
      "78560    Write a story about someone who finds somethin...\n",
      "78561    Write a story about someone who finds somethin...\n",
      "78562    Write a story about someone who finds somethin...\n",
      "78563    Write a story about someone who finds somethin...\n",
      "                               ...                        \n",
      "78644    Write a story about someone who finds somethin...\n",
      "78645    Write a story about someone who finds somethin...\n",
      "78646    Write a story about someone who finds somethin...\n",
      "78647    Write a story about someone who finds somethin...\n",
      "78648    Write a story about someone who finds somethin...\n",
      "Name: prompt, Length: 90, dtype: object\n",
      "prompt_0852\n",
      "24702    End your story with a character standing in th...\n",
      "24703    End your story with a character standing in th...\n",
      "24704    End your story with a character standing in th...\n",
      "24705    End your story with a character standing in th...\n",
      "24706    End your story with a character standing in th...\n",
      "                               ...                        \n",
      "24799    End your story with a character standing in th...\n",
      "24800    End your story with a character standing in th...\n",
      "24801    End your story with a character standing in th...\n",
      "24802    End your story with a character standing in th...\n",
      "24803    End your story with a character standing in th...\n",
      "Name: prompt, Length: 102, dtype: object\n",
      "prompt_1153\n",
      "66372    Write a story about someone who's haunted by t...\n",
      "66373    Write a story about someone who's haunted by t...\n",
      "66374    Write a story about someone who's haunted by t...\n",
      "66375    Write a story about someone who's haunted by t...\n",
      "66376    Write a story about someone who's haunted by t...\n",
      "                               ...                        \n",
      "66621    Write a story about someone who's haunted by t...\n",
      "66622    Write a story about someone who's haunted by t...\n",
      "66623    Write a story about someone who's haunted by t...\n",
      "66624    Write a story about someone who's haunted by t...\n",
      "66625    Write a story about someone who's haunted by t...\n",
      "Name: prompt, Length: 254, dtype: object\n",
      "prompt_1167\n",
      "69390    Write a story that takes place in a waiting room.\n",
      "69391    Write a story that takes place in a waiting room.\n",
      "69392    Write a story that takes place in a waiting room.\n",
      "69393    Write a story that takes place in a waiting room.\n",
      "69394    Write a story that takes place in a waiting room.\n",
      "                               ...                        \n",
      "69643    Write a story that takes place in a waiting room.\n",
      "69644    Write a story that takes place in a waiting room.\n",
      "69645    Write a story that takes place in a waiting room.\n",
      "69646    Write a story that takes place in a waiting room.\n",
      "69647    Write a story that takes place in a waiting room.\n",
      "Name: prompt, Length: 258, dtype: object\n",
      "prompt_0901\n",
      "28473    Frame your story as an adult recalling the eve...\n",
      "28474    Frame your story as an adult recalling the eve...\n",
      "28475    Frame your story as an adult recalling the eve...\n",
      "28476    Frame your story as an adult recalling the eve...\n",
      "28477    Frame your story as an adult recalling the eve...\n",
      "                               ...                        \n",
      "28714    Frame your story as an adult recalling the eve...\n",
      "28715    Frame your story as an adult recalling the eve...\n",
      "28716    Frame your story as an adult recalling the eve...\n",
      "28717    Frame your story as an adult recalling the eve...\n",
      "28718    Frame your story as an adult recalling the eve...\n",
      "Name: prompt, Length: 246, dtype: object\n",
      "prompt_1160\n",
      "68088    Write a story about a person experiencing pre-...\n",
      "68089    Write a story about a person experiencing pre-...\n",
      "68090    Write a story about a person experiencing pre-...\n",
      "68091    Write a story about a person experiencing pre-...\n",
      "68092    Write a story about a person experiencing pre-...\n",
      "                               ...                        \n",
      "68245    Write a story about a person experiencing pre-...\n",
      "68246    Write a story about a person experiencing pre-...\n",
      "68247    Write a story about a person experiencing pre-...\n",
      "68248    Write a story about a person experiencing pre-...\n",
      "68249    Write a story about a person experiencing pre-...\n",
      "Name: prompt, Length: 162, dtype: object\n",
      "prompt_0712\n",
      "18628    Start your story with the narrator or a charac...\n",
      "18629    Start your story with the narrator or a charac...\n",
      "18630    Start your story with the narrator or a charac...\n",
      "18631    Start your story with the narrator or a charac...\n",
      "18632    Start your story with the narrator or a charac...\n",
      "                               ...                        \n",
      "18726    Start your story with the narrator or a charac...\n",
      "18727    Start your story with the narrator or a charac...\n",
      "18728    Start your story with the narrator or a charac...\n",
      "18729    Start your story with the narrator or a charac...\n",
      "18730    Start your story with the narrator or a charac...\n",
      "Name: prompt, Length: 103, dtype: object\n",
      "prompt_0901\n",
      "28473    Frame your story as an adult recalling the eve...\n",
      "28474    Frame your story as an adult recalling the eve...\n",
      "28475    Frame your story as an adult recalling the eve...\n",
      "28476    Frame your story as an adult recalling the eve...\n",
      "28477    Frame your story as an adult recalling the eve...\n",
      "                               ...                        \n",
      "28714    Frame your story as an adult recalling the eve...\n",
      "28715    Frame your story as an adult recalling the eve...\n",
      "28716    Frame your story as an adult recalling the eve...\n",
      "28717    Frame your story as an adult recalling the eve...\n",
      "28718    Frame your story as an adult recalling the eve...\n",
      "Name: prompt, Length: 246, dtype: object\n",
      "prompt_1120\n",
      "60854    Write about a character who’s stuck in an elev...\n",
      "60855    Write about a character who’s stuck in an elev...\n",
      "60856    Write about a character who’s stuck in an elev...\n",
      "60857    Write about a character who’s stuck in an elev...\n",
      "60858    Write about a character who’s stuck in an elev...\n",
      "                               ...                        \n",
      "61103    Write about a character who’s stuck in an elev...\n",
      "61104    Write about a character who’s stuck in an elev...\n",
      "61105    Write about a character who’s stuck in an elev...\n",
      "61106    Write about a character who’s stuck in an elev...\n",
      "61107    Write about a character who’s stuck in an elev...\n",
      "Name: prompt, Length: 254, dtype: object\n",
      "prompt_1089\n",
      "56188    Write about a family attempting to hide their ...\n",
      "56189    Write about a family attempting to hide their ...\n",
      "56190    Write about a family attempting to hide their ...\n",
      "56191    Write about a family attempting to hide their ...\n",
      "56192    Write about a family attempting to hide their ...\n",
      "                               ...                        \n",
      "56305    Write about a family attempting to hide their ...\n",
      "56306    Write about a family attempting to hide their ...\n",
      "56307    Write about a family attempting to hide their ...\n",
      "56308    Write about a family attempting to hide their ...\n",
      "56309    Write about a family attempting to hide their ...\n",
      "Name: prompt, Length: 122, dtype: object\n",
      "prompt_1087\n",
      "55709    Start your story with two characters deciding ...\n",
      "55710    Start your story with two characters deciding ...\n",
      "55711    Start your story with two characters deciding ...\n",
      "55712    Start your story with two characters deciding ...\n",
      "55713    Start your story with two characters deciding ...\n",
      "                               ...                        \n",
      "56027    Start your story with two characters deciding ...\n",
      "56028    Start your story with two characters deciding ...\n",
      "56029    Start your story with two characters deciding ...\n",
      "56030    Start your story with two characters deciding ...\n",
      "56031    Start your story with two characters deciding ...\n",
      "Name: prompt, Length: 323, dtype: object\n",
      "prompt_1111\n",
      "59385    Write a post-apocalyptic story that features z...\n",
      "59386    Write a post-apocalyptic story that features z...\n",
      "59387    Write a post-apocalyptic story that features z...\n",
      "59388    Write a post-apocalyptic story that features z...\n",
      "59389    Write a post-apocalyptic story that features z...\n",
      "                               ...                        \n",
      "59559    Write a post-apocalyptic story that features z...\n",
      "59560    Write a post-apocalyptic story that features z...\n",
      "59561    Write a post-apocalyptic story that features z...\n",
      "59562    Write a post-apocalyptic story that features z...\n",
      "59563    Write a post-apocalyptic story that features z...\n",
      "Name: prompt, Length: 179, dtype: object\n",
      "prompt_0937\n",
      "34660    Start your story with someone being presented ...\n",
      "34661    Start your story with someone being presented ...\n",
      "34662    Start your story with someone being presented ...\n",
      "34663    Start your story with someone being presented ...\n",
      "34664    Start your story with someone being presented ...\n",
      "                               ...                        \n",
      "34876    Start your story with someone being presented ...\n",
      "34877    Start your story with someone being presented ...\n",
      "34878    Start your story with someone being presented ...\n",
      "34879    Start your story with someone being presented ...\n",
      "34880    Start your story with someone being presented ...\n",
      "Name: prompt, Length: 221, dtype: object\n",
      "prompt_1024\n",
      "46870    Write a story set in the summer, when suddenly...\n",
      "46871    Write a story set in the summer, when suddenly...\n",
      "46872    Write a story set in the summer, when suddenly...\n",
      "46873    Write a story set in the summer, when suddenly...\n",
      "46874    Write a story set in the summer, when suddenly...\n",
      "                               ...                        \n",
      "47029    Write a story set in the summer, when suddenly...\n",
      "47030    Write a story set in the summer, when suddenly...\n",
      "47031    Write a story set in the summer, when suddenly...\n",
      "47032    Write a story set in the summer, when suddenly...\n",
      "47033    Write a story set in the summer, when suddenly...\n",
      "Name: prompt, Length: 164, dtype: object\n",
      "prompt_1091\n",
      "56566    Write a ghost story where there’s more going o...\n",
      "56567    Write a ghost story where there’s more going o...\n",
      "56568    Write a ghost story where there’s more going o...\n",
      "56569    Write a ghost story where there’s more going o...\n",
      "56570    Write a ghost story where there’s more going o...\n",
      "                               ...                        \n",
      "56789    Write a ghost story where there’s more going o...\n",
      "56790    Write a ghost story where there’s more going o...\n",
      "56791    Write a ghost story where there’s more going o...\n",
      "56792    Write a ghost story where there’s more going o...\n",
      "56793    Write a ghost story where there’s more going o...\n",
      "Name: prompt, Length: 228, dtype: object\n",
      "prompt_0968\n",
      "39373    Start the story with the absence of a sensory ...\n",
      "39374    Start the story with the absence of a sensory ...\n",
      "39375    Start the story with the absence of a sensory ...\n",
      "39376    Start the story with the absence of a sensory ...\n",
      "39377    Start the story with the absence of a sensory ...\n",
      "39378    Start the story with the absence of a sensory ...\n",
      "39379    Start the story with the absence of a sensory ...\n",
      "39380    Start the story with the absence of a sensory ...\n",
      "39381    Start the story with the absence of a sensory ...\n",
      "39382    Start the story with the absence of a sensory ...\n",
      "39383    Start the story with the absence of a sensory ...\n",
      "39384    Start the story with the absence of a sensory ...\n",
      "39385    Start the story with the absence of a sensory ...\n",
      "39386    Start the story with the absence of a sensory ...\n",
      "39387    Start the story with the absence of a sensory ...\n",
      "39388    Start the story with the absence of a sensory ...\n",
      "39389    Start the story with the absence of a sensory ...\n",
      "39390    Start the story with the absence of a sensory ...\n",
      "39391    Start the story with the absence of a sensory ...\n",
      "39392    Start the story with the absence of a sensory ...\n",
      "39393    Start the story with the absence of a sensory ...\n",
      "39394    Start the story with the absence of a sensory ...\n",
      "39395    Start the story with the absence of a sensory ...\n",
      "39396    Start the story with the absence of a sensory ...\n",
      "39397    Start the story with the absence of a sensory ...\n",
      "39398    Start the story with the absence of a sensory ...\n",
      "39399    Start the story with the absence of a sensory ...\n",
      "39400    Start the story with the absence of a sensory ...\n",
      "39401    Start the story with the absence of a sensory ...\n",
      "39402    Start the story with the absence of a sensory ...\n",
      "39403    Start the story with the absence of a sensory ...\n",
      "39404    Start the story with the absence of a sensory ...\n",
      "39405    Start the story with the absence of a sensory ...\n",
      "39406    Start the story with the absence of a sensory ...\n",
      "39407    Start the story with the absence of a sensory ...\n",
      "39408    Start the story with the absence of a sensory ...\n",
      "39409    Start the story with the absence of a sensory ...\n",
      "39410    Start the story with the absence of a sensory ...\n",
      "39411    Start the story with the absence of a sensory ...\n",
      "39412    Start the story with the absence of a sensory ...\n",
      "39413    Start the story with the absence of a sensory ...\n",
      "39414    Start the story with the absence of a sensory ...\n",
      "39415    Start the story with the absence of a sensory ...\n",
      "39416    Start the story with the absence of a sensory ...\n",
      "39417    Start the story with the absence of a sensory ...\n",
      "39418    Start the story with the absence of a sensory ...\n",
      "39419    Start the story with the absence of a sensory ...\n",
      "39420    Start the story with the absence of a sensory ...\n",
      "39421    Start the story with the absence of a sensory ...\n",
      "39422    Start the story with the absence of a sensory ...\n",
      "39423    Start the story with the absence of a sensory ...\n",
      "39424    Start the story with the absence of a sensory ...\n",
      "39425    Start the story with the absence of a sensory ...\n",
      "Name: prompt, dtype: object\n",
      "prompt_0117\n",
      "6151    Write about someone stuck in an endless cycle ...\n",
      "6152    Write about someone stuck in an endless cycle ...\n",
      "6153    Write about someone stuck in an endless cycle ...\n",
      "6154    Write about someone stuck in an endless cycle ...\n",
      "6155    Write about someone stuck in an endless cycle ...\n",
      "                              ...                        \n",
      "6217    Write about someone stuck in an endless cycle ...\n",
      "6218    Write about someone stuck in an endless cycle ...\n",
      "6219    Write about someone stuck in an endless cycle ...\n",
      "6220    Write about someone stuck in an endless cycle ...\n",
      "6221    Write about someone stuck in an endless cycle ...\n",
      "Name: prompt, Length: 71, dtype: object\n",
      "prompt_1142\n",
      "64471    Write a story about high school sweethearts co...\n",
      "64472    Write a story about high school sweethearts co...\n",
      "64473    Write a story about high school sweethearts co...\n",
      "64474    Write a story about high school sweethearts co...\n",
      "64475    Write a story about high school sweethearts co...\n",
      "                               ...                        \n",
      "64753    Write a story about high school sweethearts co...\n",
      "64754    Write a story about high school sweethearts co...\n",
      "64755    Write a story about high school sweethearts co...\n",
      "64756    Write a story about high school sweethearts co...\n",
      "64757    Write a story about high school sweethearts co...\n",
      "Name: prompt, Length: 287, dtype: object\n",
      "prompt_0146\n",
      "7765    Write a story entirely of dialogue. Nothing bu...\n",
      "7766    Write a story entirely of dialogue. Nothing bu...\n",
      "7767    Write a story entirely of dialogue. Nothing bu...\n",
      "7768    Write a story entirely of dialogue. Nothing bu...\n",
      "7769    Write a story entirely of dialogue. Nothing bu...\n",
      "                              ...                        \n",
      "8044    Write a story entirely of dialogue. Nothing bu...\n",
      "8045    Write a story entirely of dialogue. Nothing bu...\n",
      "8046    Write a story entirely of dialogue. Nothing bu...\n",
      "8047    Write a story entirely of dialogue. Nothing bu...\n",
      "8048    Write a story entirely of dialogue. Nothing bu...\n",
      "Name: prompt, Length: 284, dtype: object\n",
      "prompt_1005\n",
      "43705    Write about a first date that surprises both p...\n",
      "43706    Write about a first date that surprises both p...\n",
      "43707    Write about a first date that surprises both p...\n",
      "43708    Write about a first date that surprises both p...\n",
      "43709    Write about a first date that surprises both p...\n",
      "                               ...                        \n",
      "43862    Write about a first date that surprises both p...\n",
      "43863    Write about a first date that surprises both p...\n",
      "43864    Write about a first date that surprises both p...\n",
      "43865    Write about a first date that surprises both p...\n",
      "43866    Write about a first date that surprises both p...\n",
      "Name: prompt, Length: 162, dtype: object\n",
      "prompt_1084\n",
      "55202    Write about a vampire or werewolf who moves in...\n",
      "55203    Write about a vampire or werewolf who moves in...\n",
      "55204    Write about a vampire or werewolf who moves in...\n",
      "55205    Write about a vampire or werewolf who moves in...\n",
      "55206    Write about a vampire or werewolf who moves in...\n",
      "                               ...                        \n",
      "55359    Write about a vampire or werewolf who moves in...\n",
      "55360    Write about a vampire or werewolf who moves in...\n",
      "55361    Write about a vampire or werewolf who moves in...\n",
      "55362    Write about a vampire or werewolf who moves in...\n",
      "55363    Write about a vampire or werewolf who moves in...\n",
      "Name: prompt, Length: 162, dtype: object\n",
      "prompt_1172\n",
      "69960    Write about someone who has a superpower.\n",
      "69961    Write about someone who has a superpower.\n",
      "69962    Write about someone who has a superpower.\n",
      "69963    Write about someone who has a superpower.\n",
      "69964    Write about someone who has a superpower.\n",
      "                           ...                    \n",
      "70292    Write about someone who has a superpower.\n",
      "70293    Write about someone who has a superpower.\n",
      "70294    Write about someone who has a superpower.\n",
      "70295    Write about someone who has a superpower.\n",
      "70296    Write about someone who has a superpower.\n",
      "Name: prompt, Length: 337, dtype: object\n",
      "prompt_0944\n",
      "35521    Write your story about two characters tidying ...\n",
      "35522    Write your story about two characters tidying ...\n",
      "35523    Write your story about two characters tidying ...\n",
      "35524    Write your story about two characters tidying ...\n",
      "35525    Write your story about two characters tidying ...\n",
      "                               ...                        \n",
      "35668    Write your story about two characters tidying ...\n",
      "35669    Write your story about two characters tidying ...\n",
      "35670    Write your story about two characters tidying ...\n",
      "35671    Write your story about two characters tidying ...\n",
      "35672    Write your story about two characters tidying ...\n",
      "Name: prompt, Length: 152, dtype: object\n",
      "prompt_1005\n",
      "43705    Write about a first date that surprises both p...\n",
      "43706    Write about a first date that surprises both p...\n",
      "43707    Write about a first date that surprises both p...\n",
      "43708    Write about a first date that surprises both p...\n",
      "43709    Write about a first date that surprises both p...\n",
      "                               ...                        \n",
      "43862    Write about a first date that surprises both p...\n",
      "43863    Write about a first date that surprises both p...\n",
      "43864    Write about a first date that surprises both p...\n",
      "43865    Write about a first date that surprises both p...\n",
      "43866    Write about a first date that surprises both p...\n",
      "Name: prompt, Length: 162, dtype: object\n",
      "prompt_1027\n",
      "47435    Write about two people going sledding for the ...\n",
      "47436    Write about two people going sledding for the ...\n",
      "47437    Write about two people going sledding for the ...\n",
      "47438    Write about two people going sledding for the ...\n",
      "47439    Write about two people going sledding for the ...\n",
      "                               ...                        \n",
      "47561    Write about two people going sledding for the ...\n",
      "47562    Write about two people going sledding for the ...\n",
      "47563    Write about two people going sledding for the ...\n",
      "47564    Write about two people going sledding for the ...\n",
      "47565    Write about two people going sledding for the ...\n",
      "Name: prompt, Length: 131, dtype: object\n",
      "prompt_0985\n",
      "40711    Start your story with the line, “That’s the th...\n",
      "40712    Start your story with the line, “That’s the th...\n",
      "40713    Start your story with the line, “That’s the th...\n",
      "40714    Start your story with the line, “That’s the th...\n",
      "40715    Start your story with the line, “That’s the th...\n",
      "                               ...                        \n",
      "40992    Start your story with the line, “That’s the th...\n",
      "40993    Start your story with the line, “That’s the th...\n",
      "40994    Start your story with the line, “That’s the th...\n",
      "40995    Start your story with the line, “That’s the th...\n",
      "40996    Start your story with the line, “That’s the th...\n",
      "Name: prompt, Length: 286, dtype: object\n",
      "prompt_0985\n",
      "40711    Start your story with the line, “That’s the th...\n",
      "40712    Start your story with the line, “That’s the th...\n",
      "40713    Start your story with the line, “That’s the th...\n",
      "40714    Start your story with the line, “That’s the th...\n",
      "40715    Start your story with the line, “That’s the th...\n",
      "                               ...                        \n",
      "40992    Start your story with the line, “That’s the th...\n",
      "40993    Start your story with the line, “That’s the th...\n",
      "40994    Start your story with the line, “That’s the th...\n",
      "40995    Start your story with the line, “That’s the th...\n",
      "40996    Start your story with the line, “That’s the th...\n",
      "Name: prompt, Length: 286, dtype: object\n",
      "prompt_1069\n",
      "52899    Start your story with the line, “This was supp...\n",
      "52900    Start your story with the line, “This was supp...\n",
      "52901    Start your story with the line, “This was supp...\n",
      "52902    Start your story with the line, “This was supp...\n",
      "52903    Start your story with the line, “This was supp...\n",
      "                               ...                        \n",
      "53026    Start your story with the line, “This was supp...\n",
      "53027    Start your story with the line, “This was supp...\n",
      "53028    Start your story with the line, “This was supp...\n",
      "53029    Start your story with the line, “This was supp...\n",
      "53030    Start your story with the line, “This was supp...\n",
      "Name: prompt, Length: 132, dtype: object\n",
      "prompt_1150\n",
      "65712    You thought he was dead, but there he is, righ...\n",
      "65713    You thought he was dead, but there he is, righ...\n",
      "65714    You thought he was dead, but there he is, righ...\n",
      "65715    You thought he was dead, but there he is, righ...\n",
      "65716    You thought he was dead, but there he is, righ...\n",
      "                               ...                        \n",
      "66127    You thought he was dead, but there he is, righ...\n",
      "66128    You thought he was dead, but there he is, righ...\n",
      "66129    You thought he was dead, but there he is, righ...\n",
      "66130    You thought he was dead, but there he is, righ...\n",
      "66131    You thought he was dead, but there he is, righ...\n",
      "Name: prompt, Length: 420, dtype: object\n",
      "prompt_0953\n",
      "37637    Write about a character stumbling upon a libra...\n",
      "37638    Write about a character stumbling upon a libra...\n",
      "37639    Write about a character stumbling upon a libra...\n",
      "37640    Write about a character stumbling upon a libra...\n",
      "37641    Write about a character stumbling upon a libra...\n",
      "                               ...                        \n",
      "37842    Write about a character stumbling upon a libra...\n",
      "37843    Write about a character stumbling upon a libra...\n",
      "37844    Write about a character stumbling upon a libra...\n",
      "37845    Write about a character stumbling upon a libra...\n",
      "37846    Write about a character stumbling upon a libra...\n",
      "Name: prompt, Length: 210, dtype: object\n",
      "prompt_0919\n",
      "31610    Set your story on (or in) a winding river.\n",
      "31611    Set your story on (or in) a winding river.\n",
      "31612    Set your story on (or in) a winding river.\n",
      "31613    Set your story on (or in) a winding river.\n",
      "31614    Set your story on (or in) a winding river.\n",
      "                            ...                    \n",
      "31730    Set your story on (or in) a winding river.\n",
      "31731    Set your story on (or in) a winding river.\n",
      "31732    Set your story on (or in) a winding river.\n",
      "31733    Set your story on (or in) a winding river.\n",
      "31734    Set your story on (or in) a winding river.\n",
      "Name: prompt, Length: 125, dtype: object\n",
      "prompt_0952\n",
      "36994    Write a story that begins in the light and end...\n",
      "36995    Write a story that begins in the light and end...\n",
      "36996    Write a story that begins in the light and end...\n",
      "36997    Write a story that begins in the light and end...\n",
      "36998    Write a story that begins in the light and end...\n",
      "                               ...                        \n",
      "37376    Write a story that begins in the light and end...\n",
      "37377    Write a story that begins in the light and end...\n",
      "37378    Write a story that begins in the light and end...\n",
      "37379    Write a story that begins in the light and end...\n",
      "37380    Write a story that begins in the light and end...\n",
      "Name: prompt, Length: 387, dtype: object\n",
      "prompt_1160\n",
      "68088    Write a story about a person experiencing pre-...\n",
      "68089    Write a story about a person experiencing pre-...\n",
      "68090    Write a story about a person experiencing pre-...\n",
      "68091    Write a story about a person experiencing pre-...\n",
      "68092    Write a story about a person experiencing pre-...\n",
      "                               ...                        \n",
      "68245    Write a story about a person experiencing pre-...\n",
      "68246    Write a story about a person experiencing pre-...\n",
      "68247    Write a story about a person experiencing pre-...\n",
      "68248    Write a story about a person experiencing pre-...\n",
      "68249    Write a story about a person experiencing pre-...\n",
      "Name: prompt, Length: 162, dtype: object\n",
      "prompt_1176\n",
      "71095    \"Just say it,\" you silently reminded yourself....\n",
      "71096    \"Just say it,\" you silently reminded yourself....\n",
      "71097    \"Just say it,\" you silently reminded yourself....\n",
      "71098    \"Just say it,\" you silently reminded yourself....\n",
      "71099    \"Just say it,\" you silently reminded yourself....\n",
      "                               ...                        \n",
      "71311    \"Just say it,\" you silently reminded yourself....\n",
      "71312    \"Just say it,\" you silently reminded yourself....\n",
      "71313    \"Just say it,\" you silently reminded yourself....\n",
      "71314    \"Just say it,\" you silently reminded yourself....\n",
      "71315    \"Just say it,\" you silently reminded yourself....\n",
      "Name: prompt, Length: 221, dtype: object\n",
      "prompt_1116\n",
      "60115    Write about a character arriving in a place un...\n",
      "60116    Write about a character arriving in a place un...\n",
      "60117    Write about a character arriving in a place un...\n",
      "60118    Write about a character arriving in a place un...\n",
      "60119    Write about a character arriving in a place un...\n",
      "                               ...                        \n",
      "60311    Write about a character arriving in a place un...\n",
      "60312    Write about a character arriving in a place un...\n",
      "60313    Write about a character arriving in a place un...\n",
      "60314    Write about a character arriving in a place un...\n",
      "60315    Write about a character arriving in a place un...\n",
      "Name: prompt, Length: 201, dtype: object\n",
      "prompt_1115\n",
      "60046    Set your story in a place with extreme weather...\n",
      "60047    Set your story in a place with extreme weather...\n",
      "60048    Set your story in a place with extreme weather...\n",
      "60049    Set your story in a place with extreme weather...\n",
      "60050    Set your story in a place with extreme weather...\n",
      "                               ...                        \n",
      "60110    Set your story in a place with extreme weather...\n",
      "60111    Set your story in a place with extreme weather...\n",
      "60112    Set your story in a place with extreme weather...\n",
      "60113    Set your story in a place with extreme weather...\n",
      "60114    Set your story in a place with extreme weather...\n",
      "Name: prompt, Length: 69, dtype: object\n",
      "prompt_1165\n",
      "68891    Write a story about waiting — but don't reveal...\n",
      "68892    Write a story about waiting — but don't reveal...\n",
      "68893    Write a story about waiting — but don't reveal...\n",
      "68894    Write a story about waiting — but don't reveal...\n",
      "68895    Write a story about waiting — but don't reveal...\n",
      "                               ...                        \n",
      "69211    Write a story about waiting — but don't reveal...\n",
      "69212    Write a story about waiting — but don't reveal...\n",
      "69213    Write a story about waiting — but don't reveal...\n",
      "69214    Write a story about waiting — but don't reveal...\n",
      "69215    Write a story about waiting — but don't reveal...\n",
      "Name: prompt, Length: 325, dtype: object\n",
      "prompt_1077\n",
      "54378    Write about a pirate captain obsessed with fin...\n",
      "54379    Write about a pirate captain obsessed with fin...\n",
      "54380    Write about a pirate captain obsessed with fin...\n",
      "54381    Write about a pirate captain obsessed with fin...\n",
      "54382    Write about a pirate captain obsessed with fin...\n",
      "                               ...                        \n",
      "54518    Write about a pirate captain obsessed with fin...\n",
      "54519    Write about a pirate captain obsessed with fin...\n",
      "54520    Write about a pirate captain obsessed with fin...\n",
      "54521    Write about a pirate captain obsessed with fin...\n",
      "54522    Write about a pirate captain obsessed with fin...\n",
      "Name: prompt, Length: 145, dtype: object\n",
      "prompt_0931\n",
      "33661    Start your story with the arrival of a strange...\n",
      "33662    Start your story with the arrival of a strange...\n",
      "33663    Start your story with the arrival of a strange...\n",
      "33664    Start your story with the arrival of a strange...\n",
      "33665    Start your story with the arrival of a strange...\n",
      "                               ...                        \n",
      "33846    Start your story with the arrival of a strange...\n",
      "33847    Start your story with the arrival of a strange...\n",
      "33848    Start your story with the arrival of a strange...\n",
      "33849    Start your story with the arrival of a strange...\n",
      "33850    Start your story with the arrival of a strange...\n",
      "Name: prompt, Length: 190, dtype: object\n",
      "prompt_0960\n",
      "38500    Set your story in a world living with the cons...\n",
      "38501    Set your story in a world living with the cons...\n",
      "38502    Set your story in a world living with the cons...\n",
      "38503    Set your story in a world living with the cons...\n",
      "38504    Set your story in a world living with the cons...\n",
      "                               ...                        \n",
      "38630    Set your story in a world living with the cons...\n",
      "38631    Set your story in a world living with the cons...\n",
      "38632    Set your story in a world living with the cons...\n",
      "38633    Set your story in a world living with the cons...\n",
      "38634    Set your story in a world living with the cons...\n",
      "Name: prompt, Length: 135, dtype: object\n",
      "prompt_1189\n",
      "72488    Write a story that starts with a character-rev...\n",
      "72489    Write a story that starts with a character-rev...\n",
      "72490    Write a story that starts with a character-rev...\n",
      "72491    Write a story that starts with a character-rev...\n",
      "72492    Write a story that starts with a character-rev...\n",
      "                               ...                        \n",
      "72626    Write a story that starts with a character-rev...\n",
      "72627    Write a story that starts with a character-rev...\n",
      "72628    Write a story that starts with a character-rev...\n",
      "72629    Write a story that starts with a character-rev...\n",
      "72630    Write a story that starts with a character-rev...\n",
      "Name: prompt, Length: 143, dtype: object\n",
      "prompt_0146\n",
      "7765    Write a story entirely of dialogue. Nothing bu...\n",
      "7766    Write a story entirely of dialogue. Nothing bu...\n",
      "7767    Write a story entirely of dialogue. Nothing bu...\n",
      "7768    Write a story entirely of dialogue. Nothing bu...\n",
      "7769    Write a story entirely of dialogue. Nothing bu...\n",
      "                              ...                        \n",
      "8044    Write a story entirely of dialogue. Nothing bu...\n",
      "8045    Write a story entirely of dialogue. Nothing bu...\n",
      "8046    Write a story entirely of dialogue. Nothing bu...\n",
      "8047    Write a story entirely of dialogue. Nothing bu...\n",
      "8048    Write a story entirely of dialogue. Nothing bu...\n",
      "Name: prompt, Length: 284, dtype: object\n",
      "prompt_0932\n",
      "33393    Write a story about strangers becoming friends...\n",
      "33394    Write a story about strangers becoming friends...\n",
      "33395    Write a story about strangers becoming friends...\n",
      "33396    Write a story about strangers becoming friends...\n",
      "33397    Write a story about strangers becoming friends...\n",
      "                               ...                        \n",
      "33656    Write a story about strangers becoming friends...\n",
      "33657    Write a story about strangers becoming friends...\n",
      "33658    Write a story about strangers becoming friends...\n",
      "33659    Write a story about strangers becoming friends...\n",
      "33660    Write a story about strangers becoming friends...\n",
      "Name: prompt, Length: 268, dtype: object\n",
      "prompt_1142\n",
      "64471    Write a story about high school sweethearts co...\n",
      "64472    Write a story about high school sweethearts co...\n",
      "64473    Write a story about high school sweethearts co...\n",
      "64474    Write a story about high school sweethearts co...\n",
      "64475    Write a story about high school sweethearts co...\n",
      "                               ...                        \n",
      "64753    Write a story about high school sweethearts co...\n",
      "64754    Write a story about high school sweethearts co...\n",
      "64755    Write a story about high school sweethearts co...\n",
      "64756    Write a story about high school sweethearts co...\n",
      "64757    Write a story about high school sweethearts co...\n",
      "Name: prompt, Length: 287, dtype: object\n",
      "prompt_1096\n",
      "57385    Start your story with the line, “By the time I...\n",
      "57386    Start your story with the line, “By the time I...\n",
      "57387    Start your story with the line, “By the time I...\n",
      "57388    Start your story with the line, “By the time I...\n",
      "57389    Start your story with the line, “By the time I...\n",
      "                               ...                        \n",
      "57706    Start your story with the line, “By the time I...\n",
      "57707    Start your story with the line, “By the time I...\n",
      "57708    Start your story with the line, “By the time I...\n",
      "57709    Start your story with the line, “By the time I...\n",
      "57710    Start your story with the line, “By the time I...\n",
      "Name: prompt, Length: 326, dtype: object\n",
      "prompt_1177\n",
      "70881    As you check your mail, you notice a letter th...\n",
      "70882    As you check your mail, you notice a letter th...\n",
      "70883    As you check your mail, you notice a letter th...\n",
      "70884    As you check your mail, you notice a letter th...\n",
      "70885    As you check your mail, you notice a letter th...\n",
      "                               ...                        \n",
      "71090    As you check your mail, you notice a letter th...\n",
      "71091    As you check your mail, you notice a letter th...\n",
      "71092    As you check your mail, you notice a letter th...\n",
      "71093    As you check your mail, you notice a letter th...\n",
      "71094    As you check your mail, you notice a letter th...\n",
      "Name: prompt, Length: 214, dtype: object\n",
      "prompt_1168\n",
      "69648    Write a story that features a protagonist with...\n",
      "69649    Write a story that features a protagonist with...\n",
      "69650    Write a story that features a protagonist with...\n",
      "69651    Write a story that features a protagonist with...\n",
      "69652    Write a story that features a protagonist with...\n",
      "                               ...                        \n",
      "69756    Write a story that features a protagonist with...\n",
      "69757    Write a story that features a protagonist with...\n",
      "69758    Write a story that features a protagonist with...\n",
      "69759    Write a story that features a protagonist with...\n",
      "69760    Write a story that features a protagonist with...\n",
      "Name: prompt, Length: 113, dtype: object\n",
      "prompt_1158\n",
      "67634    Write a story about a summer afternoon spent i...\n",
      "67635    Write a story about a summer afternoon spent i...\n",
      "67636    Write a story about a summer afternoon spent i...\n",
      "67637    Write a story about a summer afternoon spent i...\n",
      "67638    Write a story about a summer afternoon spent i...\n",
      "                               ...                        \n",
      "67937    Write a story about a summer afternoon spent i...\n",
      "67938    Write a story about a summer afternoon spent i...\n",
      "67939    Write a story about a summer afternoon spent i...\n",
      "67940    Write a story about a summer afternoon spent i...\n",
      "67941    Write a story about a summer afternoon spent i...\n",
      "Name: prompt, Length: 308, dtype: object\n",
      "prompt_0937\n",
      "34660    Start your story with someone being presented ...\n",
      "34661    Start your story with someone being presented ...\n",
      "34662    Start your story with someone being presented ...\n",
      "34663    Start your story with someone being presented ...\n",
      "34664    Start your story with someone being presented ...\n",
      "                               ...                        \n",
      "34876    Start your story with someone being presented ...\n",
      "34877    Start your story with someone being presented ...\n",
      "34878    Start your story with someone being presented ...\n",
      "34879    Start your story with someone being presented ...\n",
      "34880    Start your story with someone being presented ...\n",
      "Name: prompt, Length: 221, dtype: object\n",
      "prompt_0044\n",
      "2338    Write about a person who constantly has to put...\n",
      "2339    Write about a person who constantly has to put...\n",
      "2340    Write about a person who constantly has to put...\n",
      "2341    Write about a person who constantly has to put...\n",
      "2342    Write about a person who constantly has to put...\n",
      "                              ...                        \n",
      "2416    Write about a person who constantly has to put...\n",
      "2417    Write about a person who constantly has to put...\n",
      "2418    Write about a person who constantly has to put...\n",
      "2419    Write about a person who constantly has to put...\n",
      "2420    Write about a person who constantly has to put...\n",
      "Name: prompt, Length: 83, dtype: object\n",
      "prompt_1142\n",
      "64471    Write a story about high school sweethearts co...\n",
      "64472    Write a story about high school sweethearts co...\n",
      "64473    Write a story about high school sweethearts co...\n",
      "64474    Write a story about high school sweethearts co...\n",
      "64475    Write a story about high school sweethearts co...\n",
      "                               ...                        \n",
      "64753    Write a story about high school sweethearts co...\n",
      "64754    Write a story about high school sweethearts co...\n",
      "64755    Write a story about high school sweethearts co...\n",
      "64756    Write a story about high school sweethearts co...\n",
      "64757    Write a story about high school sweethearts co...\n",
      "Name: prompt, Length: 287, dtype: object\n",
      "prompt_1167\n",
      "69390    Write a story that takes place in a waiting room.\n",
      "69391    Write a story that takes place in a waiting room.\n",
      "69392    Write a story that takes place in a waiting room.\n",
      "69393    Write a story that takes place in a waiting room.\n",
      "69394    Write a story that takes place in a waiting room.\n",
      "                               ...                        \n",
      "69643    Write a story that takes place in a waiting room.\n",
      "69644    Write a story that takes place in a waiting room.\n",
      "69645    Write a story that takes place in a waiting room.\n",
      "69646    Write a story that takes place in a waiting room.\n",
      "69647    Write a story that takes place in a waiting room.\n",
      "Name: prompt, Length: 258, dtype: object\n",
      "prompt_1274\n",
      "81742    Write a short story that ends with a twist.\n",
      "81743    Write a short story that ends with a twist.\n",
      "81744    Write a short story that ends with a twist.\n",
      "81745    Write a short story that ends with a twist.\n",
      "81746    Write a short story that ends with a twist.\n",
      "                            ...                     \n",
      "81894    Write a short story that ends with a twist.\n",
      "81895    Write a short story that ends with a twist.\n",
      "81896    Write a short story that ends with a twist.\n",
      "81897    Write a short story that ends with a twist.\n",
      "81898    Write a short story that ends with a twist.\n",
      "Name: prompt, Length: 157, dtype: object\n",
      "prompt_1017\n",
      "45613    Write about someone who decides it’s time to c...\n",
      "45614    Write about someone who decides it’s time to c...\n",
      "45615    Write about someone who decides it’s time to c...\n",
      "45616    Write about someone who decides it’s time to c...\n",
      "45617    Write about someone who decides it’s time to c...\n",
      "                               ...                        \n",
      "45873    Write about someone who decides it’s time to c...\n",
      "45874    Write about someone who decides it’s time to c...\n",
      "45875    Write about someone who decides it’s time to c...\n",
      "45876    Write about someone who decides it’s time to c...\n",
      "45877    Write about someone who decides it’s time to c...\n",
      "Name: prompt, Length: 265, dtype: object\n",
      "prompt_0638\n",
      "15353    Write a story featuring an element of time-tra...\n",
      "15354    Write a story featuring an element of time-tra...\n",
      "15355    Write a story featuring an element of time-tra...\n",
      "15356    Write a story featuring an element of time-tra...\n",
      "15357    Write a story featuring an element of time-tra...\n",
      "                               ...                        \n",
      "15416    Write a story featuring an element of time-tra...\n",
      "15417    Write a story featuring an element of time-tra...\n",
      "15418    Write a story featuring an element of time-tra...\n",
      "15419    Write a story featuring an element of time-tra...\n",
      "15420    Write a story featuring an element of time-tra...\n",
      "Name: prompt, Length: 68, dtype: object\n",
      "prompt_1197\n",
      "73639    Write a story about transformation.\n",
      "73640    Write a story about transformation.\n",
      "73641    Write a story about transformation.\n",
      "73642    Write a story about transformation.\n",
      "73643    Write a story about transformation.\n",
      "                        ...                 \n",
      "73807    Write a story about transformation.\n",
      "73808    Write a story about transformation.\n",
      "73809    Write a story about transformation.\n",
      "73810    Write a story about transformation.\n",
      "73811    Write a story about transformation.\n",
      "Name: prompt, Length: 173, dtype: object\n",
      "prompt_0935\n",
      "34213    A character stands in front of two doors. Writ...\n",
      "34214    A character stands in front of two doors. Writ...\n",
      "34215    A character stands in front of two doors. Writ...\n",
      "34216    A character stands in front of two doors. Writ...\n",
      "34217    A character stands in front of two doors. Writ...\n",
      "                               ...                        \n",
      "34448    A character stands in front of two doors. Writ...\n",
      "34449    A character stands in front of two doors. Writ...\n",
      "34450    A character stands in front of two doors. Writ...\n",
      "34451    A character stands in front of two doors. Writ...\n",
      "34452    A character stands in front of two doors. Writ...\n",
      "Name: prompt, Length: 240, dtype: object\n",
      "prompt_0620\n",
      "14572    Write a story where the law plays an important...\n",
      "14573    Write a story where the law plays an important...\n",
      "14574    Write a story where the law plays an important...\n",
      "14575    Write a story where the law plays an important...\n",
      "14576    Write a story where the law plays an important...\n",
      "14577    Write a story where the law plays an important...\n",
      "14578    Write a story where the law plays an important...\n",
      "14579    Write a story where the law plays an important...\n",
      "14580    Write a story where the law plays an important...\n",
      "14581    Write a story where the law plays an important...\n",
      "14582    Write a story where the law plays an important...\n",
      "14583    Write a story where the law plays an important...\n",
      "14584    Write a story where the law plays an important...\n",
      "14585    Write a story where the law plays an important...\n",
      "14586    Write a story where the law plays an important...\n",
      "14587    Write a story where the law plays an important...\n",
      "14588    Write a story where the law plays an important...\n",
      "14589    Write a story where the law plays an important...\n",
      "14590    Write a story where the law plays an important...\n",
      "14591    Write a story where the law plays an important...\n",
      "14592    Write a story where the law plays an important...\n",
      "14593    Write a story where the law plays an important...\n",
      "14594    Write a story where the law plays an important...\n",
      "14595    Write a story where the law plays an important...\n",
      "Name: prompt, dtype: object\n",
      "prompt_0923\n",
      "32047    Set your story within a window of opportunity,...\n",
      "32048    Set your story within a window of opportunity,...\n",
      "32049    Set your story within a window of opportunity,...\n",
      "32050    Set your story within a window of opportunity,...\n",
      "32051    Set your story within a window of opportunity,...\n",
      "                               ...                        \n",
      "32189    Set your story within a window of opportunity,...\n",
      "32190    Set your story within a window of opportunity,...\n",
      "32191    Set your story within a window of opportunity,...\n",
      "32192    Set your story within a window of opportunity,...\n",
      "32193    Set your story within a window of opportunity,...\n",
      "Name: prompt, Length: 147, dtype: object\n",
      "prompt_0912\n",
      "29891    Start or end your story with two characters si...\n",
      "29892    Start or end your story with two characters si...\n",
      "29893    Start or end your story with two characters si...\n",
      "29894    Start or end your story with two characters si...\n",
      "29895    Start or end your story with two characters si...\n",
      "                               ...                        \n",
      "30153    Start or end your story with two characters si...\n",
      "30154    Start or end your story with two characters si...\n",
      "30155    Start or end your story with two characters si...\n",
      "30156    Start or end your story with two characters si...\n",
      "30157    Start or end your story with two characters si...\n",
      "Name: prompt, Length: 267, dtype: object\n",
      "prompt_1032\n",
      "48007    Write a story told exclusively through dialogue.\n",
      "48008    Write a story told exclusively through dialogue.\n",
      "48009    Write a story told exclusively through dialogue.\n",
      "48010    Write a story told exclusively through dialogue.\n",
      "48011    Write a story told exclusively through dialogue.\n",
      "                               ...                       \n",
      "48356    Write a story told exclusively through dialogue.\n",
      "48357    Write a story told exclusively through dialogue.\n",
      "48358    Write a story told exclusively through dialogue.\n",
      "48359    Write a story told exclusively through dialogue.\n",
      "48360    Write a story told exclusively through dialogue.\n",
      "Name: prompt, Length: 354, dtype: object\n",
      "prompt_1225\n",
      "77452    Write a story that takes place in the woods.\n",
      "77453    Write a story that takes place in the woods.\n",
      "77454    Write a story that takes place in the woods.\n",
      "77455    Write a story that takes place in the woods.\n",
      "77456    Write a story that takes place in the woods.\n",
      "                             ...                     \n",
      "77628    Write a story that takes place in the woods.\n",
      "77629    Write a story that takes place in the woods.\n",
      "77630    Write a story that takes place in the woods.\n",
      "77631    Write a story that takes place in the woods.\n",
      "77632    Write a story that takes place in the woods.\n",
      "Name: prompt, Length: 181, dtype: object\n",
      "prompt_0958\n",
      "38022    Start your story with someone sitting on a cro...\n",
      "38023    Start your story with someone sitting on a cro...\n",
      "38024    Start your story with someone sitting on a cro...\n",
      "38025    Start your story with someone sitting on a cro...\n",
      "38026    Start your story with someone sitting on a cro...\n",
      "                               ...                        \n",
      "38279    Start your story with someone sitting on a cro...\n",
      "38280    Start your story with someone sitting on a cro...\n",
      "38281    Start your story with someone sitting on a cro...\n",
      "38282    Start your story with someone sitting on a cro...\n",
      "38283    Start your story with someone sitting on a cro...\n",
      "Name: prompt, Length: 262, dtype: object\n",
      "prompt_1013\n",
      "44965    Write about a “found family” who are finally a...\n",
      "44966    Write about a “found family” who are finally a...\n",
      "44967    Write about a “found family” who are finally a...\n",
      "44968    Write about a “found family” who are finally a...\n",
      "44969    Write about a “found family” who are finally a...\n",
      "                               ...                        \n",
      "45061    Write about a “found family” who are finally a...\n",
      "45062    Write about a “found family” who are finally a...\n",
      "45063    Write about a “found family” who are finally a...\n",
      "45064    Write about a “found family” who are finally a...\n",
      "45065    Write about a “found family” who are finally a...\n",
      "Name: prompt, Length: 101, dtype: object\n",
      "prompt_0998\n",
      "42817    Write about an android just trying to blend in...\n",
      "42818    Write about an android just trying to blend in...\n",
      "42819    Write about an android just trying to blend in...\n",
      "42820    Write about an android just trying to blend in...\n",
      "42821    Write about an android just trying to blend in...\n",
      "                               ...                        \n",
      "43037    Write about an android just trying to blend in...\n",
      "43038    Write about an android just trying to blend in...\n",
      "43039    Write about an android just trying to blend in...\n",
      "43040    Write about an android just trying to blend in...\n",
      "43041    Write about an android just trying to blend in...\n",
      "Name: prompt, Length: 225, dtype: object\n",
      "prompt_0986\n",
      "40997    Write a story that takes place in the same bui...\n",
      "40998    Write a story that takes place in the same bui...\n",
      "40999    Write a story that takes place in the same bui...\n",
      "41000    Write a story that takes place in the same bui...\n",
      "41001    Write a story that takes place in the same bui...\n",
      "                               ...                        \n",
      "41148    Write a story that takes place in the same bui...\n",
      "41149    Write a story that takes place in the same bui...\n",
      "41150    Write a story that takes place in the same bui...\n",
      "41151    Write a story that takes place in the same bui...\n",
      "41152    Write a story that takes place in the same bui...\n",
      "Name: prompt, Length: 156, dtype: object\n",
      "prompt_0980\n",
      "40220    Write a fairy tale about someone who can commu...\n",
      "40221    Write a fairy tale about someone who can commu...\n",
      "40222    Write a fairy tale about someone who can commu...\n",
      "40223    Write a fairy tale about someone who can commu...\n",
      "40224    Write a fairy tale about someone who can commu...\n",
      "                               ...                        \n",
      "40384    Write a fairy tale about someone who can commu...\n",
      "40385    Write a fairy tale about someone who can commu...\n",
      "40386    Write a fairy tale about someone who can commu...\n",
      "40387    Write a fairy tale about someone who can commu...\n",
      "40388    Write a fairy tale about someone who can commu...\n",
      "Name: prompt, Length: 169, dtype: object\n",
      "prompt_1168\n",
      "69648    Write a story that features a protagonist with...\n",
      "69649    Write a story that features a protagonist with...\n",
      "69650    Write a story that features a protagonist with...\n",
      "69651    Write a story that features a protagonist with...\n",
      "69652    Write a story that features a protagonist with...\n",
      "                               ...                        \n",
      "69756    Write a story that features a protagonist with...\n",
      "69757    Write a story that features a protagonist with...\n",
      "69758    Write a story that features a protagonist with...\n",
      "69759    Write a story that features a protagonist with...\n",
      "69760    Write a story that features a protagonist with...\n",
      "Name: prompt, Length: 113, dtype: object\n",
      "prompt_1105\n",
      "58423    Write about a character who smells something f...\n",
      "58424    Write about a character who smells something f...\n",
      "58425    Write about a character who smells something f...\n",
      "58426    Write about a character who smells something f...\n",
      "58427    Write about a character who smells something f...\n",
      "                               ...                        \n",
      "58639    Write about a character who smells something f...\n",
      "58640    Write about a character who smells something f...\n",
      "58641    Write about a character who smells something f...\n",
      "58642    Write about a character who smells something f...\n",
      "58643    Write about a character who smells something f...\n",
      "Name: prompt, Length: 221, dtype: object\n",
      "prompt_1159\n",
      "67942    Write a story told entirely through one chase ...\n",
      "67943    Write a story told entirely through one chase ...\n",
      "67944    Write a story told entirely through one chase ...\n",
      "67945    Write a story told entirely through one chase ...\n",
      "67946    Write a story told entirely through one chase ...\n",
      "                               ...                        \n",
      "68083    Write a story told entirely through one chase ...\n",
      "68084    Write a story told entirely through one chase ...\n",
      "68085    Write a story told entirely through one chase ...\n",
      "68086    Write a story told entirely through one chase ...\n",
      "68087    Write a story told entirely through one chase ...\n",
      "Name: prompt, Length: 146, dtype: object\n",
      "prompt_1186\n",
      "72443    Write a story about activism.\n",
      "72444    Write a story about activism.\n",
      "72445    Write a story about activism.\n",
      "72446    Write a story about activism.\n",
      "72447    Write a story about activism.\n",
      "72448    Write a story about activism.\n",
      "72449    Write a story about activism.\n",
      "72450    Write a story about activism.\n",
      "72451    Write a story about activism.\n",
      "72452    Write a story about activism.\n",
      "72453    Write a story about activism.\n",
      "72454    Write a story about activism.\n",
      "72455    Write a story about activism.\n",
      "72456    Write a story about activism.\n",
      "72457    Write a story about activism.\n",
      "72458    Write a story about activism.\n",
      "72459    Write a story about activism.\n",
      "72460    Write a story about activism.\n",
      "72461    Write a story about activism.\n",
      "72462    Write a story about activism.\n",
      "72463    Write a story about activism.\n",
      "72464    Write a story about activism.\n",
      "72465    Write a story about activism.\n",
      "72466    Write a story about activism.\n",
      "72467    Write a story about activism.\n",
      "72468    Write a story about activism.\n",
      "72469    Write a story about activism.\n",
      "72470    Write a story about activism.\n",
      "72471    Write a story about activism.\n",
      "72472    Write a story about activism.\n",
      "72473    Write a story about activism.\n",
      "72474    Write a story about activism.\n",
      "72475    Write a story about activism.\n",
      "72476    Write a story about activism.\n",
      "72477    Write a story about activism.\n",
      "72478    Write a story about activism.\n",
      "72479    Write a story about activism.\n",
      "72480    Write a story about activism.\n",
      "72481    Write a story about activism.\n",
      "72482    Write a story about activism.\n",
      "72483    Write a story about activism.\n",
      "72484    Write a story about activism.\n",
      "72485    Write a story about activism.\n",
      "72486    Write a story about activism.\n",
      "72487    Write a story about activism.\n",
      "Name: prompt, dtype: object\n",
      "prompt_1114\n",
      "59814    Write a story that feels lonely, despite being...\n",
      "59815    Write a story that feels lonely, despite being...\n",
      "59816    Write a story that feels lonely, despite being...\n",
      "59817    Write a story that feels lonely, despite being...\n",
      "59818    Write a story that feels lonely, despite being...\n",
      "                               ...                        \n",
      "60041    Write a story that feels lonely, despite being...\n",
      "60042    Write a story that feels lonely, despite being...\n",
      "60043    Write a story that feels lonely, despite being...\n",
      "60044    Write a story that feels lonely, despite being...\n",
      "60045    Write a story that feels lonely, despite being...\n",
      "Name: prompt, Length: 232, dtype: object\n",
      "prompt_0701\n",
      "18109    Write a story about a librarian that doesn’t f...\n",
      "18110    Write a story about a librarian that doesn’t f...\n",
      "18111    Write a story about a librarian that doesn’t f...\n",
      "18112    Write a story about a librarian that doesn’t f...\n",
      "18113    Write a story about a librarian that doesn’t f...\n",
      "                               ...                        \n",
      "18168    Write a story about a librarian that doesn’t f...\n",
      "18169    Write a story about a librarian that doesn’t f...\n",
      "18170    Write a story about a librarian that doesn’t f...\n",
      "18171    Write a story about a librarian that doesn’t f...\n",
      "18172    Write a story about a librarian that doesn’t f...\n",
      "Name: prompt, Length: 64, dtype: object\n",
      "prompt_1061\n",
      "52046    Write about a character who everyone thinks is...\n",
      "52047    Write about a character who everyone thinks is...\n",
      "52048    Write about a character who everyone thinks is...\n",
      "52049    Write about a character who everyone thinks is...\n",
      "52050    Write about a character who everyone thinks is...\n",
      "                               ...                        \n",
      "52222    Write about a character who everyone thinks is...\n",
      "52223    Write about a character who everyone thinks is...\n",
      "52224    Write about a character who everyone thinks is...\n",
      "52225    Write about a character who everyone thinks is...\n",
      "52226    Write about a character who everyone thinks is...\n",
      "Name: prompt, Length: 181, dtype: object\n",
      "prompt_0925\n",
      "32333    Write a story that involves a magic window — o...\n",
      "32334    Write a story that involves a magic window — o...\n",
      "32335    Write a story that involves a magic window — o...\n",
      "32336    Write a story that involves a magic window — o...\n",
      "32337    Write a story that involves a magic window — o...\n",
      "                               ...                        \n",
      "32484    Write a story that involves a magic window — o...\n",
      "32485    Write a story that involves a magic window — o...\n",
      "32486    Write a story that involves a magic window — o...\n",
      "32487    Write a story that involves a magic window — o...\n",
      "32488    Write a story that involves a magic window — o...\n",
      "Name: prompt, Length: 156, dtype: object\n",
      "prompt_1018\n",
      "45878    Write about someone who keeps picking up diffe...\n",
      "45879    Write about someone who keeps picking up diffe...\n",
      "45880    Write about someone who keeps picking up diffe...\n",
      "45881    Write about someone who keeps picking up diffe...\n",
      "45882    Write about someone who keeps picking up diffe...\n",
      "                               ...                        \n",
      "46050    Write about someone who keeps picking up diffe...\n",
      "46051    Write about someone who keeps picking up diffe...\n",
      "46052    Write about someone who keeps picking up diffe...\n",
      "46053    Write about someone who keeps picking up diffe...\n",
      "46054    Write about someone who keeps picking up diffe...\n",
      "Name: prompt, Length: 177, dtype: object\n",
      "prompt_1094\n",
      "57046    Write a story from the perspective of a bird m...\n",
      "57047    Write a story from the perspective of a bird m...\n",
      "57048    Write a story from the perspective of a bird m...\n",
      "57049    Write a story from the perspective of a bird m...\n",
      "57050    Write a story from the perspective of a bird m...\n",
      "                               ...                        \n",
      "57171    Write a story from the perspective of a bird m...\n",
      "57172    Write a story from the perspective of a bird m...\n",
      "57173    Write a story from the perspective of a bird m...\n",
      "57174    Write a story from the perspective of a bird m...\n",
      "57175    Write a story from the perspective of a bird m...\n",
      "Name: prompt, Length: 130, dtype: object\n",
      "prompt_1167\n",
      "69390    Write a story that takes place in a waiting room.\n",
      "69391    Write a story that takes place in a waiting room.\n",
      "69392    Write a story that takes place in a waiting room.\n",
      "69393    Write a story that takes place in a waiting room.\n",
      "69394    Write a story that takes place in a waiting room.\n",
      "                               ...                        \n",
      "69643    Write a story that takes place in a waiting room.\n",
      "69644    Write a story that takes place in a waiting room.\n",
      "69645    Write a story that takes place in a waiting room.\n",
      "69646    Write a story that takes place in a waiting room.\n",
      "69647    Write a story that takes place in a waiting room.\n",
      "Name: prompt, Length: 258, dtype: object\n",
      "prompt_1017\n",
      "45613    Write about someone who decides it’s time to c...\n",
      "45614    Write about someone who decides it’s time to c...\n",
      "45615    Write about someone who decides it’s time to c...\n",
      "45616    Write about someone who decides it’s time to c...\n",
      "45617    Write about someone who decides it’s time to c...\n",
      "                               ...                        \n",
      "45873    Write about someone who decides it’s time to c...\n",
      "45874    Write about someone who decides it’s time to c...\n",
      "45875    Write about someone who decides it’s time to c...\n",
      "45876    Write about someone who decides it’s time to c...\n",
      "45877    Write about someone who decides it’s time to c...\n",
      "Name: prompt, Length: 265, dtype: object\n",
      "prompt_1072\n",
      "53481    Start your story with two characters watching ...\n",
      "53482    Start your story with two characters watching ...\n",
      "53483    Start your story with two characters watching ...\n",
      "53484    Start your story with two characters watching ...\n",
      "53485    Start your story with two characters watching ...\n",
      "                               ...                        \n",
      "53780    Start your story with two characters watching ...\n",
      "53781    Start your story with two characters watching ...\n",
      "53782    Start your story with two characters watching ...\n",
      "53783    Start your story with two characters watching ...\n",
      "53784    Start your story with two characters watching ...\n",
      "Name: prompt, Length: 304, dtype: object\n",
      "prompt_0796\n",
      "22465    Write about a character who always wears a mas...\n",
      "22466    Write about a character who always wears a mas...\n",
      "22467    Write about a character who always wears a mas...\n",
      "22468    Write about a character who always wears a mas...\n",
      "22469    Write about a character who always wears a mas...\n",
      "22470    Write about a character who always wears a mas...\n",
      "22471    Write about a character who always wears a mas...\n",
      "22472    Write about a character who always wears a mas...\n",
      "22473    Write about a character who always wears a mas...\n",
      "22474    Write about a character who always wears a mas...\n",
      "22475    Write about a character who always wears a mas...\n",
      "22476    Write about a character who always wears a mas...\n",
      "22477    Write about a character who always wears a mas...\n",
      "22478    Write about a character who always wears a mas...\n",
      "22479    Write about a character who always wears a mas...\n",
      "22480    Write about a character who always wears a mas...\n",
      "22481    Write about a character who always wears a mas...\n",
      "22482    Write about a character who always wears a mas...\n",
      "22483    Write about a character who always wears a mas...\n",
      "22484    Write about a character who always wears a mas...\n",
      "22485    Write about a character who always wears a mas...\n",
      "22486    Write about a character who always wears a mas...\n",
      "22487    Write about a character who always wears a mas...\n",
      "22488    Write about a character who always wears a mas...\n",
      "22489    Write about a character who always wears a mas...\n",
      "22490    Write about a character who always wears a mas...\n",
      "22491    Write about a character who always wears a mas...\n",
      "22492    Write about a character who always wears a mas...\n",
      "22493    Write about a character who always wears a mas...\n",
      "22494    Write about a character who always wears a mas...\n",
      "22495    Write about a character who always wears a mas...\n",
      "22496    Write about a character who always wears a mas...\n",
      "22497    Write about a character who always wears a mas...\n",
      "22498    Write about a character who always wears a mas...\n",
      "22499    Write about a character who always wears a mas...\n",
      "22500    Write about a character who always wears a mas...\n",
      "22501    Write about a character who always wears a mas...\n",
      "22502    Write about a character who always wears a mas...\n",
      "22503    Write about a character who always wears a mas...\n",
      "22504    Write about a character who always wears a mas...\n",
      "22505    Write about a character who always wears a mas...\n",
      "22506    Write about a character who always wears a mas...\n",
      "22507    Write about a character who always wears a mas...\n",
      "22508    Write about a character who always wears a mas...\n",
      "22509    Write about a character who always wears a mas...\n",
      "22510    Write about a character who always wears a mas...\n",
      "22511    Write about a character who always wears a mas...\n",
      "22512    Write about a character who always wears a mas...\n",
      "22513    Write about a character who always wears a mas...\n",
      "Name: prompt, dtype: object\n",
      "prompt_0862\n",
      "25026    Set your story in a roadside diner.\n",
      "25027    Set your story in a roadside diner.\n",
      "25028    Set your story in a roadside diner.\n",
      "25029    Set your story in a roadside diner.\n",
      "25030    Set your story in a roadside diner.\n",
      "25031    Set your story in a roadside diner.\n",
      "25032    Set your story in a roadside diner.\n",
      "25033    Set your story in a roadside diner.\n",
      "25034    Set your story in a roadside diner.\n",
      "25035    Set your story in a roadside diner.\n",
      "25036    Set your story in a roadside diner.\n",
      "25037    Set your story in a roadside diner.\n",
      "25038    Set your story in a roadside diner.\n",
      "25039    Set your story in a roadside diner.\n",
      "25040    Set your story in a roadside diner.\n",
      "25041    Set your story in a roadside diner.\n",
      "25042    Set your story in a roadside diner.\n",
      "25043    Set your story in a roadside diner.\n",
      "25044    Set your story in a roadside diner.\n",
      "25045    Set your story in a roadside diner.\n",
      "25046    Set your story in a roadside diner.\n",
      "25047    Set your story in a roadside diner.\n",
      "25048    Set your story in a roadside diner.\n",
      "25049    Set your story in a roadside diner.\n",
      "25050    Set your story in a roadside diner.\n",
      "25051    Set your story in a roadside diner.\n",
      "25052    Set your story in a roadside diner.\n",
      "25053    Set your story in a roadside diner.\n",
      "25054    Set your story in a roadside diner.\n",
      "25055    Set your story in a roadside diner.\n",
      "25056    Set your story in a roadside diner.\n",
      "25057    Set your story in a roadside diner.\n",
      "25058    Set your story in a roadside diner.\n",
      "25059    Set your story in a roadside diner.\n",
      "25060    Set your story in a roadside diner.\n",
      "25061    Set your story in a roadside diner.\n",
      "25062    Set your story in a roadside diner.\n",
      "25063    Set your story in a roadside diner.\n",
      "25064    Set your story in a roadside diner.\n",
      "25065    Set your story in a roadside diner.\n",
      "25066    Set your story in a roadside diner.\n",
      "25067    Set your story in a roadside diner.\n",
      "25068    Set your story in a roadside diner.\n",
      "25069    Set your story in a roadside diner.\n",
      "25070    Set your story in a roadside diner.\n",
      "25071    Set your story in a roadside diner.\n",
      "25072    Set your story in a roadside diner.\n",
      "25073    Set your story in a roadside diner.\n",
      "25074    Set your story in a roadside diner.\n",
      "25075    Set your story in a roadside diner.\n",
      "25076    Set your story in a roadside diner.\n",
      "25077    Set your story in a roadside diner.\n",
      "25078    Set your story in a roadside diner.\n",
      "25079    Set your story in a roadside diner.\n",
      "25080    Set your story in a roadside diner.\n",
      "25081    Set your story in a roadside diner.\n",
      "25082    Set your story in a roadside diner.\n",
      "25083    Set your story in a roadside diner.\n",
      "25084    Set your story in a roadside diner.\n",
      "Name: prompt, dtype: object\n",
      "prompt_1010\n",
      "44580    Start your story with someone entering a museu...\n",
      "44581    Start your story with someone entering a museu...\n",
      "44582    Start your story with someone entering a museu...\n",
      "44583    Start your story with someone entering a museu...\n",
      "44584    Start your story with someone entering a museu...\n",
      "44585    Start your story with someone entering a museu...\n",
      "44586    Start your story with someone entering a museu...\n",
      "44587    Start your story with someone entering a museu...\n",
      "44588    Start your story with someone entering a museu...\n",
      "44589    Start your story with someone entering a museu...\n",
      "44590    Start your story with someone entering a museu...\n",
      "44591    Start your story with someone entering a museu...\n",
      "44592    Start your story with someone entering a museu...\n",
      "44593    Start your story with someone entering a museu...\n",
      "44594    Start your story with someone entering a museu...\n",
      "44595    Start your story with someone entering a museu...\n",
      "44596    Start your story with someone entering a museu...\n",
      "44597    Start your story with someone entering a museu...\n",
      "44598    Start your story with someone entering a museu...\n",
      "44599    Start your story with someone entering a museu...\n",
      "44600    Start your story with someone entering a museu...\n",
      "44601    Start your story with someone entering a museu...\n",
      "44602    Start your story with someone entering a museu...\n",
      "44603    Start your story with someone entering a museu...\n",
      "44604    Start your story with someone entering a museu...\n",
      "44605    Start your story with someone entering a museu...\n",
      "44606    Start your story with someone entering a museu...\n",
      "44607    Start your story with someone entering a museu...\n",
      "44608    Start your story with someone entering a museu...\n",
      "44609    Start your story with someone entering a museu...\n",
      "44610    Start your story with someone entering a museu...\n",
      "44611    Start your story with someone entering a museu...\n",
      "44612    Start your story with someone entering a museu...\n",
      "44613    Start your story with someone entering a museu...\n",
      "44614    Start your story with someone entering a museu...\n",
      "44615    Start your story with someone entering a museu...\n",
      "44616    Start your story with someone entering a museu...\n",
      "44617    Start your story with someone entering a museu...\n",
      "44618    Start your story with someone entering a museu...\n",
      "44619    Start your story with someone entering a museu...\n",
      "44620    Start your story with someone entering a museu...\n",
      "Name: prompt, dtype: object\n",
      "prompt_1069\n",
      "52899    Start your story with the line, “This was supp...\n",
      "52900    Start your story with the line, “This was supp...\n",
      "52901    Start your story with the line, “This was supp...\n",
      "52902    Start your story with the line, “This was supp...\n",
      "52903    Start your story with the line, “This was supp...\n",
      "                               ...                        \n",
      "53026    Start your story with the line, “This was supp...\n",
      "53027    Start your story with the line, “This was supp...\n",
      "53028    Start your story with the line, “This was supp...\n",
      "53029    Start your story with the line, “This was supp...\n",
      "53030    Start your story with the line, “This was supp...\n",
      "Name: prompt, Length: 132, dtype: object\n",
      "prompt_1155\n",
      "66626    Write a story that involves a mystery — it doe...\n",
      "66627    Write a story that involves a mystery — it doe...\n",
      "66628    Write a story that involves a mystery — it doe...\n",
      "66629    Write a story that involves a mystery — it doe...\n",
      "66630    Write a story that involves a mystery — it doe...\n",
      "                               ...                        \n",
      "66832    Write a story that involves a mystery — it doe...\n",
      "66833    Write a story that involves a mystery — it doe...\n",
      "66834    Write a story that involves a mystery — it doe...\n",
      "66835    Write a story that involves a mystery — it doe...\n",
      "66836    Write a story that involves a mystery — it doe...\n",
      "Name: prompt, Length: 211, dtype: object\n",
      "prompt_1025\n",
      "47034    Write about someone who gets stuck in their wo...\n",
      "47035    Write about someone who gets stuck in their wo...\n",
      "47036    Write about someone who gets stuck in their wo...\n",
      "47037    Write about someone who gets stuck in their wo...\n",
      "47038    Write about someone who gets stuck in their wo...\n",
      "                               ...                        \n",
      "47193    Write about someone who gets stuck in their wo...\n",
      "47194    Write about someone who gets stuck in their wo...\n",
      "47195    Write about someone who gets stuck in their wo...\n",
      "47196    Write about someone who gets stuck in their wo...\n",
      "47197    Write about someone who gets stuck in their wo...\n",
      "Name: prompt, Length: 164, dtype: object\n",
      "prompt_1157\n",
      "67225    Write a story that begins and ends with someon...\n",
      "67226    Write a story that begins and ends with someon...\n",
      "67227    Write a story that begins and ends with someon...\n",
      "67228    Write a story that begins and ends with someon...\n",
      "67229    Write a story that begins and ends with someon...\n",
      "                               ...                        \n",
      "67629    Write a story that begins and ends with someon...\n",
      "67630    Write a story that begins and ends with someon...\n",
      "67631    Write a story that begins and ends with someon...\n",
      "67632    Write a story that begins and ends with someon...\n",
      "67633    Write a story that begins and ends with someon...\n",
      "Name: prompt, Length: 409, dtype: object\n",
      "prompt_0992\n",
      "41767    Write a story that spans exactly a year and ta...\n",
      "41768    Write a story that spans exactly a year and ta...\n",
      "41769    Write a story that spans exactly a year and ta...\n",
      "41770    Write a story that spans exactly a year and ta...\n",
      "41771    Write a story that spans exactly a year and ta...\n",
      "                               ...                        \n",
      "41974    Write a story that spans exactly a year and ta...\n",
      "41975    Write a story that spans exactly a year and ta...\n",
      "41976    Write a story that spans exactly a year and ta...\n",
      "41977    Write a story that spans exactly a year and ta...\n",
      "41978    Write a story that spans exactly a year and ta...\n",
      "Name: prompt, Length: 212, dtype: object\n",
      "prompt_0118\n",
      "6222    Write a story about someone trying to reinvent...\n",
      "6223    Write a story about someone trying to reinvent...\n",
      "6224    Write a story about someone trying to reinvent...\n",
      "6225    Write a story about someone trying to reinvent...\n",
      "6226    Write a story about someone trying to reinvent...\n",
      "6227    Write a story about someone trying to reinvent...\n",
      "6228    Write a story about someone trying to reinvent...\n",
      "6229    Write a story about someone trying to reinvent...\n",
      "6230    Write a story about someone trying to reinvent...\n",
      "6231    Write a story about someone trying to reinvent...\n",
      "6232    Write a story about someone trying to reinvent...\n",
      "6233    Write a story about someone trying to reinvent...\n",
      "6234    Write a story about someone trying to reinvent...\n",
      "6235    Write a story about someone trying to reinvent...\n",
      "6236    Write a story about someone trying to reinvent...\n",
      "6237    Write a story about someone trying to reinvent...\n",
      "6238    Write a story about someone trying to reinvent...\n",
      "6239    Write a story about someone trying to reinvent...\n",
      "6240    Write a story about someone trying to reinvent...\n",
      "6241    Write a story about someone trying to reinvent...\n",
      "6242    Write a story about someone trying to reinvent...\n",
      "6243    Write a story about someone trying to reinvent...\n",
      "6244    Write a story about someone trying to reinvent...\n",
      "6245    Write a story about someone trying to reinvent...\n",
      "6246    Write a story about someone trying to reinvent...\n",
      "6247    Write a story about someone trying to reinvent...\n",
      "6248    Write a story about someone trying to reinvent...\n",
      "6249    Write a story about someone trying to reinvent...\n",
      "6250    Write a story about someone trying to reinvent...\n",
      "6251    Write a story about someone trying to reinvent...\n",
      "6252    Write a story about someone trying to reinvent...\n",
      "6253    Write a story about someone trying to reinvent...\n",
      "6254    Write a story about someone trying to reinvent...\n",
      "6255    Write a story about someone trying to reinvent...\n",
      "6256    Write a story about someone trying to reinvent...\n",
      "6257    Write a story about someone trying to reinvent...\n",
      "6258    Write a story about someone trying to reinvent...\n",
      "6259    Write a story about someone trying to reinvent...\n",
      "Name: prompt, dtype: object\n",
      "prompt_1162\n",
      "68371    Write a story about a proposal. \n",
      "68372    Write a story about a proposal. \n",
      "68373    Write a story about a proposal. \n",
      "68374    Write a story about a proposal. \n",
      "68375    Write a story about a proposal. \n",
      "                       ...               \n",
      "68580    Write a story about a proposal. \n",
      "68581    Write a story about a proposal. \n",
      "68582    Write a story about a proposal. \n",
      "68583    Write a story about a proposal. \n",
      "68584    Write a story about a proposal. \n",
      "Name: prompt, Length: 214, dtype: object\n",
      "prompt_1172\n",
      "69960    Write about someone who has a superpower.\n",
      "69961    Write about someone who has a superpower.\n",
      "69962    Write about someone who has a superpower.\n",
      "69963    Write about someone who has a superpower.\n",
      "69964    Write about someone who has a superpower.\n",
      "                           ...                    \n",
      "70292    Write about someone who has a superpower.\n",
      "70293    Write about someone who has a superpower.\n",
      "70294    Write about someone who has a superpower.\n",
      "70295    Write about someone who has a superpower.\n",
      "70296    Write about someone who has a superpower.\n",
      "Name: prompt, Length: 337, dtype: object\n",
      "the first example of train is  prompt_id                                              prompt_1234\n",
      "story1_id                                                   rikt93\n",
      "story2_id                                                   qyd9jh\n",
      "time_lag                                                  309660.0\n",
      "least_likes                                                      8\n",
      "chosen_text      <bos><|im_start|>user\\nWrite a story about som...\n",
      "rejected_text    <bos><|im_start|>user\\nWrite a story about som...\n",
      "Name: 0, dtype: object\n"
     ]
    }
   ],
   "source": [
    "#test dataloader \n",
    "from dataloader import StoryPairDataset\n",
    "dataloader = StoryPairDataset(datapath,\n",
    "                              pairpath,\n",
    "                              tokenizer,\n",
    "                              task='rm',\n",
    "                              used_dataset_size=100,\n",
    "                              train_test_split=0.1,\n",
    "                              split_by='random',\n",
    "                              max_len=1024*4,\n",
    "                              mode='m2',\n",
    "                              max_time_window=5400,\n",
    "                              least_likes=5,\n",
    "                              margin=False)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "b11f608a-c2eb-42af-bfba-e801ee40e0ed",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.13"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}