|
import gradio as gr |
|
|
|
from convert import convert |
|
from huggingface_hub import HfApi |
|
|
|
|
|
def run(token: str, model_id: str) -> str: |
|
if token == "" or model_id == "": |
|
return """ |
|
### Invalid input 🐞 |
|
|
|
Please fill a token and model_id. |
|
""" |
|
try: |
|
|
|
|
|
api = HfApi(token=token) |
|
commit_info = convert(api=api, model_id=model_id) |
|
return 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}) |
|
""" |
|
except Exception as e: |
|
return f""" |
|
### Error 😢😢😢 |
|
|
|
{e} |
|
""" |
|
|
|
|
|
DESCRIPTION = """ |
|
The steps are the following: |
|
|
|
- Paste a read-access token from hf.co/settings/tokens. Read access is enough given that we will open a PR against the source repo. |
|
- Input a 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 opened PR 🔥 |
|
|
|
⚠️ For now only `pytorch_model.bin` files are supported but we'll extend in the future. |
|
""" |
|
|
|
demo = gr.Interface( |
|
title="Convert any model to Safetensors and open a PR", |
|
description=DESCRIPTION, |
|
allow_flagging="never", |
|
article="Check out the [Safetensors repo on GitHub](https://github.com/huggingface/safetensors)", |
|
inputs=[ |
|
gr.Text(max_lines=1, label="your_hf_token"), |
|
gr.Text(max_lines=1, label="model_id"), |
|
], |
|
outputs=[gr.Markdown(label="output")], |
|
fn=run, |
|
) |
|
|
|
demo.launch() |
|
|
