Update app.py

app.py

@@ -4,6 +4,7 @@ import networkx as nx
 import matplotlib.pyplot as plt
 import gradio as gr
 from io import BytesIO
+from PIL import Image
 
 # Define investors and their companies
 investors = {
@@ -68,7 +69,7 @@ investor_colors = {
     "Y Combinator": "black"
 }
 
-# Function to generate the graph image
+
 def generate_graph(selected_investors):
     if not selected_investors:
         selected_investors = list(investors.keys())
@@ -113,7 +114,10 @@ def generate_graph(selected_investors):
     plt.savefig(buf, format="png", bbox_inches="tight")
     plt.close()
     buf.seek(0)
-    return buf
+
+    # Convert BytesIO to PIL image
+    image = Image.open(buf)
+    return image
 
 # Define Gradio interface
 def main():