update readme.

README.md

@@ -69,6 +69,8 @@ print("==> prediciton: ", prediction)
 # output: ==> prediciton: There is one dog in the picture.
 ```
 
+More comprehenisve examples can be found in the [notebook](demo.ipynb).
+
 # License
 
 Our code and weights are released under the Creative Commons Attribution Non Commercial 4.0 [LICENSE](LICENSE.txt).

demo.ipynb

@@ -319,14 +319,14 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": 6,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "==> prediciton:  There is one dog in the picture.\n"
+      "==> prediciton:  There is one dog in the picture. \n"
      ]
     }
    ],
@@ -346,7 +346,7 @@
     "                                do_sample=False, max_new_tokens=768, top_p=None, num_beams=1,\n",
     "                                stopping_criteria = [EosListStoppingCriteria()],\n",
     "                                )\n",
-    "prediction = tokenizer.decode(generated_text[0], skip_special_tokens=True)\n",
+    "prediction = tokenizer.decode(generated_text[0], skip_special_tokens=True).split(\"<|end|>\")[0]\n",
     "print(\"==> prediciton: \", prediction)\n",
     "# output: ==> prediciton: There is one dog in the picture."
    ]