[no ci] minor edit

README.md

@@ -17,7 +17,7 @@
 ```python
 # Example: Generative AI in ktrain
 from ktrain.text.generative_ai import GenerativeAI
-model = GenerativeAI(device='cpu') # use device='cuda' if you have a good GPU!
+model = GenerativeAI() # needs at least 16GB of GPU memory
 prompt = """Extract the names of people in the supplied sentences. Here is an example:
 Sentence:Paul Newman is a great actor.
 People:

examples/text/generative_ai_example.ipynb

@@ -18,7 +18,7 @@
     "\n",
     "*ktrain* supports a Generative AI module that is currently based on an instruction-fine-tuned version of GPT-J. Think of it as a lightweight version of ChatGPT that can be run locally on your own machine. As a smaller model, it will not perform as well as GPT-4, ChatGPT, etc.  However, since it does not communicate with external APIs like OpenAI, it can be used with non-public data.\n",
     "\n",
-    "The model requires a GPU with at least 16GB of GPU memory or VRAM.  If you have less than this, you can use a CPU (provided it has at least 16GB of RAM), but output will be generated very slowly (depending on the number of CPU cores).  We will use a CPU in this example, but you should either leave blank or explicitly supply `device=cuda` if you have a GPU with at least 16GB of GPU memory."
+    "The model requires a GPU with at least 16GB of GPU memory or VRAM.  If you have less than this, you can use a CPU (provided it has at least 16GB of RAM), but output will be generated very slowly (depending on the number of CPU cores).  We will use a CPU in this example, but you should either omit the `device` parameter or explicitly supply `device=cuda` if you have a GPU with at least 16GB of GPU memory."
    ]
   },
   {
@@ -28,7 +28,7 @@
    "outputs": [],
    "source": [
     "from ktrain.text.generative_ai import GenerativeAI\n",
-    "model = GenerativeAI(device='cpu') # leave blank or use device='cuda' if you have a good GPU!"
+    "model = GenerativeAI(device='cpu') # omit device argument or use device='cuda' if you have a good GPU!"
    ]
   },
   {