barebones

app.py

@@ -1,7 +1,62 @@
 import gradio as gr
 
-def greet(name):
-    return "Hello " + name + "!!"
+def create_medusa_heads(model_id: str):
+    return "", ""
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+def run(model_id: str) -> str:
+    if model_id == "":
+        return """
+        ### Invalid input 🐞
+
+        Please fill a model_id.
+        """
+    try:
+        commit_info, errors = create_medusa_heads(model_id=model_id)
+        print("[commit_info]", commit_info)
+
+        string =  f"""
+        ### Success 🔥
+
+        Yay! This model was successfully converted and a PR was open using your token, here:
+
+        [{commit_info.pr_url}]({commit_info.pr_url})
+        """
+        if errors:
+            string += "\nErrors during conversion:\n"
+            string += "\n".join(f"Error while converting {filename}: {e}, skipped conversion" for filename, e in errors)
+        return string
+    except Exception as e:
+        return f"""
+        ### Error 😢😢😢
+
+        {e}
+        """
+
+
+DESCRIPTION = """
+The step to create [medusa](https://sites.google.com/view/medusa-llm) heads are the following:
+
+- Input a public model id from the Hub
+- Click "Submit"
+- That's it! You'll get feedback if it works or not, and if it worked, you'll get the URL of the new repo 🔥
+"""
+
+title="Create LLM medusa heads in a new repo 🐍"
+
+with gr.Blocks(title=title) as demo:
+    description = gr.Markdown(f"""# {title}""")
+    description = gr.Markdown(DESCRIPTION)
+
+    with gr.Row() as r:
+        with gr.Column() as c:
+            model_id = gr.Text(max_lines=1, label="model_id")
+            with gr.Row() as c:
+                clean = gr.ClearButton()
+                submit = gr.Button("Submit", variant="primary")
+
+        with gr.Column() as d:
+            output = gr.Markdown()
+
+    submit.click(run, inputs=[model_id], outputs=output, concurrency_limit=1)
+
+demo.queue(max_size=10).launch(show_api=True)