diff --git a/nbs/classroom_9.ipynb b/nbs/classroom_9.ipynb index 454e0e6d..b86958ed 100644 --- a/nbs/classroom_9.ipynb +++ b/nbs/classroom_9.ipynb @@ -578,8 +578,8 @@ " dialogue = dataset['test'][index]['dialogue']\n", " summary = dataset['test'][index]['summary']\n", "\n", - " start_prompt = 'Summarize this dialogue' # 'Summarize the following conversation.\\n\\n' # Summarize this dialogue.\\n\\n\n", - " end_prompt = 'Summary:' # ''\n", + " start_prompt = 'Summarize this dialogue'\n", + " end_prompt = 'Summary:' \n", " prompt = start_prompt + dialogue + end_prompt\n", "\n", " inputs = tokenizer_flan(prompt, return_tensors='pt')\n",