Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import matplotlib.pyplot as plt
+import numpy as np
 
 def create_portfolio_graph():
     # Data for the bars
@@ -23,10 +24,13 @@ def create_portfolio_graph():
     plt.savefig("/mnt/data/portfolio_chart.png")
     plt.close()
 
+    # Return the image as a NumPy array
+    return np.array(plt.imread("/mnt/data/portfolio_chart.png"))
+
 # Gradio Interface (without user input)
 iface = gr.Interface(
     fn=create_portfolio_graph,
-    outputs=gr.Image(type="file", label="Portfolio Composition Chart"),
+    outputs=gr.Image(type="numpy", label="Portfolio Composition Chart"),
     live=True,
     capture_session=True
 )
@@ -40,3 +44,4 @@ iface.launch()
 
 
 
+