Update app.py

app.py

@@ -165,7 +165,7 @@ def create_family_tree(start_model):
     plt.figtext(0.5, 0.01, "Merge Family Tree. Created by Maxime Labonne, ported to Gradio by mrfakename.\nhttps://huggingface.co/spaces/mrfakename/merge-model-tree", ha="center", fontsize=10)
 
     buf = io.BytesIO()
-    fig.savefig(buf)
+    plt.savefig(buf)
     buf.seek(0)
     img = Image.open(buf)
     return img