Update app.py

app.py

@@ -7,6 +7,7 @@ import numpy as np
 import pandas as pd
 import ast
 from collections import Counter
+from io import BytesIO
 
 
 # # Ensure GPU usage if available
@@ -187,6 +188,10 @@ test_examples = [[None, "Images/cat_dog.jpeg", promt_cat_dog],
 
 def gradio_predict(video,image, question):
     answer = predict_answer(video,image, question)
+
+    # Convert the dictionary to a DataFrame
+    df = pd.DataFrame([answer])
+
     # Convert the DataFrame to CSV
     output_buffer = BytesIO()
     df.to_csv(answer, index=False)