Update app.py

app.py

@@ -156,7 +156,7 @@ def answer_question(context, question):
         messages = [
             # {"role": "system", "content": "You are a helpful assistant who can answer questions based on the given context."},
             {"role":"system", "content": SYSTEM_MESSAGE},
-            {"role": "user", "content": f"Context: {context}\n\nQuestion: {question}"}
+            {"role": "user", "content": f"Context: {context}\n Question: {question}"}
         ]
         
         response = qa_pipeline(messages, max_new_tokens=256)[0]['generated_text']
@@ -182,10 +182,10 @@ with gr.Blocks() as iface:
             translation_output = gr.Textbox(label="Transcription/Translation", lines=10)
     
     with gr.Row():
-        with gr.Column():
-            summarize_button = gr.Button("Summarize")
-            summary_output = gr.Textbox(label="Summary", lines=3)
-            
+        # with gr.Column():
+        #     summarize_button = gr.Button("Summarize")
+        #     summary_output = gr.Textbox(label="Summary", lines=3)
+          
         with gr.Column():
             question_input = gr.Textbox(label="Ask a question about the transcription")
             answer_button = gr.Button("Get Answer")
@@ -199,12 +199,12 @@ with gr.Blocks() as iface:
         # outputs=[ASR_RESULT]
     )
     # translated_text = ''.join(item['translated'] for item in ASR_RESULT if 'translated' in item)
-    summarize_button.click(
-        summarize_text,
-        # inputs=[ASR_RESULT],
-        inputs=[translation_output],
-        outputs=[summary_output]
-    )
+    # summarize_button.click(
+    #     summarize_text,
+    #     # inputs=[ASR_RESULT],
+    #     inputs=[translation_output],
+    #     outputs=[summary_output]
+    # )
     
     answer_button.click(
         answer_question,