Change number of figures

climateqa/engine/chains/retrieve_documents.py

@@ -83,9 +83,9 @@ def _get_k_images_by_question(n_questions):
     if n_questions == 0:
         return 0
     elif n_questions == 1:
-        return 5
+        return 7
     elif n_questions == 2:
-        return 3
+        return 5
     elif n_questions == 3:
         return 2
     else: