Update app.py

app.py

@@ -35,7 +35,6 @@ with gr.Blocks() as app:
         method_choice = gr.Radio(["Regular RAG", "Graph-RAG"], label="Select Method")
         get_answer_btn = gr.Button("Get Answer")
         answer_output = gr.Textbox(label="Answer")
-        image_output = gr.Gallery(label="Relevant Images", show_label=True)
         graph_output = gr.Plot(label="Knowledge Graph Visualization")
         ask_another_question_btn = gr.Button("Ask Another Question")
         change_patient_btn = gr.Button("Set Another Patient")
@@ -53,7 +52,6 @@ with gr.Blocks() as app:
             patient_input,
             question_input,
             answer_output,
-            image_output,
             graph_output,
             ask_another_question_btn,
             change_patient_btn
@@ -63,12 +61,12 @@ with gr.Blocks() as app:
     get_answer_btn.click(
         qa_tool,
         inputs=[question_input, method_choice],
-        outputs=[answer_output, image_output, ask_another_question_btn, change_patient_btn, graph_output]
+        outputs=[answer_output, ask_another_question_btn, change_patient_btn, graph_output]
     )
 
     ask_another_question_btn.click(
         ask_another_question,
-        outputs=[question_input, answer_output, image_output, graph_output, get_answer_btn, ask_another_question_btn]
+        outputs=[question_input, answer_output, graph_output, get_answer_btn, ask_another_question_btn]
     )
 
     change_patient_btn.click(
@@ -81,11 +79,11 @@ with gr.Blocks() as app:
             change_patient_btn,
             question_input,
             answer_output,
-            image_output,
             graph_output
         ]
     )
 
+
 if __name__ == "__main__":
     logger.info("Starting the application")
     app.launch()