diff --git a/README.md b/README.md index 94d63ec..e953c11 100644 --- a/README.md +++ b/README.md @@ -102,7 +102,7 @@ messages = [ ] output = pipe(text=messages, max_new_tokens=200) -print(output[0][0]["generated_text"][-1]["content"]) +print(output[0]["generated_text"][-1]["content"]) # Okay, let's take a look! # Based on the image, the animal on the candy is a **turtle**. # You can see the shell shape and the head and legs.