Update examples/prompting/Few_shot_prompting.ipynb

Co-authored-by: Mark Daoust <markdaoust@google.com>
google-gemini · May 30, 2024 · 66cc8f7 · 66cc8f7
1 parent bb3b6f0
commit 66cc8f7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/prompting/Few_shot_prompting.ipynb b/examples/prompting/Few_shot_prompting.ipynb
@@ -194,7 +194,7 @@
         "MODEL: {\"Minsk\": \"Ukraine\", \"Monaco\": \"Monaco\", \"Maputo\": \"Mozambique\"}\n",
         "USER: I am currently in Austin, but I will be moving to Lisbon soon\n",
         "MODEL:\"\"\"\n",
-        "model.generate_content(prompt).text"
+        "model.generate_content(prompt, generation_config={'response_mime_type':'application/json'}).text"
       ]
     },
     {