Update app.py

app.py

@@ -5,7 +5,7 @@ import yaml
 import subprocess
 
 import runpod
-# import gradio as gr
+import gradio as gr
 import pandas as pd
 from jinja2 import Template
 from huggingface_hub import ModelCard, ModelCardData, HfApi, repo_info
@@ -41,7 +41,7 @@ def merge_models() -> None:
     """
     Use mergekit to create a merge.
     """
-    command = ["mergekit-yaml", "config.yaml", "merge", "--copy-tokenizer"]
+    command = ["mergekit-yaml", "config.yaml", "merge", "--copy-tokenizer", "--allow-crimes", "--out-shard-size", "1B", "--lazy-unpickle"]
 
     try:
         result = subprocess.run(command, check=True, stdout=subprocess.PIPE,
@@ -316,6 +316,7 @@ def merge_loop():
     # Evaluate model on Runpod
     create_pod(model_name, USERNAME)
 
+
 command = ["git", "clone", "-q", "https://github.com/Weyaxi/scrape-open-llm-leaderboard"]
 subprocess.run(command, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
 
@@ -336,9 +337,9 @@ title = """
   <p><em>AutoMerger selects two 7B models on top of the Open LLM Leaderboard, combine them with a merge technique, and evaluate the resulting model.</em></p>
 </div>
 """
-# with gr.Blocks() as demo:
-#     gr.Markdown(title)
-# demo.launch().launch(server_name="0.0.0.0")
+with gr.Blocks() as demo:
+    gr.Markdown(title)
+demo.launch().launch(server_name="0.0.0.0")
 
 print("Start AutoMerger...")