Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import matplotlib.pyplot as plt
 from io import BytesIO
+from PIL import Image as PILImage
 
 # Define the 44 BFI questions
 questions = [
@@ -178,7 +179,9 @@ def compute_bfi_scores(*args):
         plt.savefig(buf, format='png')
         buf.seek(0)
         plt.close(fig)
-        image = buf.getvalue()
+        # Convert bytes to PIL Image
+        image = PILImage.open(buf)
+        buf.close()
 
         markdown_output += "### **Trait Scores Visualization**\n\n"