Spaces:

mlabonne
/

model-family-tree

Running

mlabonne commited on Jan 23, 2024

Commit

bedfad8

verified ·

1 Parent(s): 1b87d33

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -168,7 +168,7 @@ with gr.Blocks() as demo:
     gr.Markdown("# 🌳 Merge Family Tree")
     model_id = gr.Textbox(label="Model ID", value="mlabonne/NeuralBeagle-7B")
     btn = gr.Button("Create tree")
-    btn.click(fn=create_family_tree, inputs=model_id, outputs=out)
     out = gr.Image()
 demo.launch()

     gr.Markdown("# 🌳 Merge Family Tree")
     model_id = gr.Textbox(label="Model ID", value="mlabonne/NeuralBeagle-7B")
     btn = gr.Button("Create tree")
     out = gr.Image()
+    btn.click(fn=create_family_tree, inputs=model_id, outputs=out)
 demo.launch()