Update app.py

app.py

@@ -186,7 +186,12 @@ test_examples = [[None, "Images/cat_dog.jpeg", promt_cat_dog],
 
 def gradio_predict(video,image, question):
     answer = predict_answer(video,image, question)
-    return answer
+    # Convert the DataFrame to CSV
+    output_buffer = BytesIO()
+    df.to_csv(answer, index=False)
+    output_buffer.seek(0)
+
+    return answer, output_buffer
 
 css = """
 #container{
@@ -221,6 +226,7 @@ with gr.Blocks(css = css) as app:
             btn = gr.Button("Annotate")
         with gr.Column():
             answer = gr.TextArea(label="Answer")
+            output_buffer = gr.File(label="Download CSV", file_count="single")
     
     
     btn.click(gradio_predict, inputs=[video,image, question], outputs=answer)