Update app.py

app.py

@@ -27,6 +27,43 @@ The fastest way to perform a model merge 🔥
 Specify a YAML configuration file (see examples below) and a HF token and this app will perform the merge and upload the merged model to your user profile.
 """
 
+MARKDOWN_ARTICLE = """
+___
+## Merge Configuration
+[Mergekit](https://github.com/arcee-ai/mergekit) configurations are YAML documents specifying the operations to perform in order to produce your merged model.
+Below are the primary elements of a configuration file:
+- `merge_method`: Specifies the method to use for merging models. See [Merge Methods](https://github.com/arcee-ai/mergekit#merge-methods) for a list.
+- `slices`: Defines slices of layers from different models to be used. This field is mutually exclusive with `models`.
+- `models`: Defines entire models to be used for merging. This field is mutually exclusive with `slices`.
+- `base_model`: Specifies the base model used in some merging methods.
+- `parameters`: Holds various parameters such as weights and densities, which can also be specified at different levels of the configuration.
+- `dtype`: Specifies the data type used for the merging operation.
+- `tokenizer_source`: Determines how to construct a tokenizer for the merged model.
+## Merge Methods
+A quick overview of the currently supported merge methods:
+| Method                                                                                       | `merge_method` value | Multi-Model | Uses base model |
+| -------------------------------------------------------------------------------------------- | -------------------- | ----------- | --------------- |
+| Linear ([Model Soups](https://arxiv.org/abs/2203.05482))                                     | `linear`             | ✅          | ❌              |
+| SLERP                                                                                        | `slerp`              | ❌          | ✅              |
+| [Task Arithmetic](https://arxiv.org/abs/2212.04089)                                          | `task_arithmetic`    | ✅          | ✅              |
+| [TIES](https://arxiv.org/abs/2306.01708)                                                     | `ties`               | ✅          | ✅              |
+| [DARE](https://arxiv.org/abs/2311.03099) [TIES](https://arxiv.org/abs/2306.01708)            | `dare_ties`          | ✅          | ✅              |
+| [DARE](https://arxiv.org/abs/2311.03099) [Task Arithmetic](https://arxiv.org/abs/2212.04089) | `dare_linear`        | ✅          | ✅              |
+| Passthrough                                                                                  | `passthrough`        | ❌          | ❌              |
+| [Model Stock](https://arxiv.org/abs/2403.19522)                                              | `model_stock`        | ✅          | ✅              |
+## Citation
+This GUI is powered by [Arcee's MergeKit](https://arxiv.org/abs/2403.13257).
+If you use it in your research, please cite the following paper:
+@article{goddard2024arcee,
+  title={Arcee's MergeKit: A Toolkit for Merging Large Language Models},
+  author={Goddard, Charles and Siriwardhana, Shamane and Ehghaghi, Malikeh and Meyers, Luke and Karpukhin, Vlad and Benedict, Brian and McQuade, Mark and Solawetz, Jacob},
+  journal={arXiv preprint arXiv:2403.13257},
+  year={2024}
+}
+
+This Space is heavily inspired by LazyMergeKit by Maxime Labonne (see [Colab](https://colab.research.google.com/drive/1obulZ1ROXHjYLn6PPZJwRR6GzgQogxxb)).
+"""
+
 examples = [[str(f)] for f in pathlib.Path("examples").glob("*.yaml")]
 COMMUNITY_HF_TOKEN = os.getenv("COMMUNITY_HF_TOKEN")
 
@@ -124,26 +161,12 @@ def generate_example_yaml(methods: List[str]) -> str:
         },
         'tokenizer_source': 'union',  # Definir el tokenizer
     }
-    # Agregar los métodos de fusión aplicados
-    example_yaml['merge_method_sequence'] = methods
+    # Ag
 
+regar la secuencia de métodos de fusión realizados
+    example_yaml['merge_method_sequence'] = methods
     return yaml.dump(example_yaml)
 
-def merge(yaml_config: str, hf_token: str, repo_name: str, profile_name: str) -> Iterable[List[Log]]:
-    runner = LogsViewRunner()
-
-    if not yaml_config:
-        yield runner.log("Empty yaml, pick an example below", level="ERROR")
-        return
-
-    try:
-        merge_config = MergeConfiguration.model_validate(yaml.safe_load(yaml_config))
-    except Exception as e:
-        yield runner.log(f"Invalid yaml {e}", level="ERROR")
-        return
-
-    yield from merge_multiple_methods(yaml_config, hf_token, repo_name, profile_name)
-
 with gr.Blocks() as demo:
     gr.Markdown(MARKDOWN_DESCRIPTION)
 
@@ -180,7 +203,7 @@ with gr.Blocks() as demo:
     )
     gr.Markdown(MARKDOWN_ARTICLE)
 
-    button.click(fn=merge, inputs=[config, token, repo_name, profile_name], outputs=[logs])
+    button.click(fn=merge_multiple_methods, inputs=[config, token, repo_name, profile_name], outputs=[logs])
 
 def _garbage_collect_every_hour():
     while True: