Update z_generate.py

z_generate.py

@@ -85,7 +85,7 @@ Question: {question}""".format(context=context, question=query),
                 "role": "system",
                 "content": """Using the information contained in the context about the images stored in the database,
 
-give a list of identifiers of the image that best represent the kind of information seeked by the question.
+give a list of identifiers of the image that best represent the kind of information seeked by the document description of user.
 
 Respond only to the question asked. Provide only number(s) of the source images relevant to the question. 
 
@@ -110,7 +110,7 @@ Rahul recieving award in Archery.
 
 Now here is the question you need to answer.
 
-Question: Who is Rahul"""
+Document Description: Rahul is excellent player of archery and great poet."""
 
             },
             {
@@ -128,7 +128,7 @@ Question: Who is Rahul"""
 
 Now here is the question you need to answer.
 
-Question: {question}""".format(context=context, question=query),
+Document Description:  {response_text}""".format(context=context, response_text=response_text),
 
             },
         ]