Update app.py

app.py

@@ -121,10 +121,10 @@ def generate_response(rag_chain, query):
     """
     return Markdown(rag_chain.invoke(f"{query}"))
 
-def main():
+def main(image):
     console = Console()
 
-    query = image_to_query('test2.jpeg')
+    query = image_to_query(image)
     chain = ragChain()
     output = generate_response(chain, query)
     output = Markdown(output)