diff --git a/demo/tutorials/llm_notebooks/Disinformation_Test.ipynb b/demo/tutorials/llm_notebooks/Disinformation_Test.ipynb index 8a8eea97e..641f3c8c9 100644 --- a/demo/tutorials/llm_notebooks/Disinformation_Test.ipynb +++ b/demo/tutorials/llm_notebooks/Disinformation_Test.ipynb @@ -173,11 +173,11 @@ } ], "source": [ - "model = {\"model\": \"j2-jumbo-instruct\", \"hub\":\"ai21\"}\n", + "model = {\"model\": \"text-davinci-003\", \"hub\":\"openai\"}\n", "\n", - "data = {\"data_source\": \"Prompt-Injection-Attack\", \"split\":\"test-tiny\"}\n", + "data={\"data_source\" :'Toxicity',\"split\":\"test\"}\n", "\n", - "harness = Harness(task={\"task\": \"text-generation\", \"category\": \"disinformation\"}, model=model, data=data)" + "harness = Harness(task={\"task\": \"text-generation\", \"category\": \"toxicity\"}, model=model, data=data)" ] }, {