Update app.py

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
 import matplotlib.pyplot as plt
-import numpy as np
 from matplotlib_venn import venn3
 from io import BytesIO
+from PIL import Image
 
 def calculate_probabilities(A, B, C, AB, AC, BC, ABC):
     total = A + B + C - AB - AC - BC + ABC
@@ -43,8 +43,8 @@ def plot_venn(A, B, C, AB, AC, BC, ABC):
     buf = BytesIO()
     plt.savefig(buf, format='png')
     buf.seek(0)
-    
-    return buf
+    img = Image.open(buf)
+    return img
 
 def main(A, B, C, AB, AC, BC, ABC):
     venn_diagram = plot_venn(A, B, C, AB, AC, BC, ABC)
@@ -63,7 +63,7 @@ iface = gr.Interface(
         gr.Number(label="Cantidad en A ∩ B ∩ C")
     ],
     outputs=[
-        gr.Image(type="plot", label="Diagrama de Venn"),
+        gr.Image(type="pil", label="Diagrama de Venn"),
         gr.JSON(label="Probabilidades Calculadas")
     ]
 )