update space

app.py

@@ -61,7 +61,7 @@ def onnx_export(token: str, model_id: str, task: str, opset: Union[int, str]) ->
 
         pr_revision = commit_info.pr_revision.replace("/", "%2F")
 
-        return f"#### Success 🔥 Yay! This model was successfully exported and a PR was open using your token, here: [{commit_info.pr_url}]({commit_info.pr_url}). If you would like to use the exported model without waiting for the PR to be approved, head to "
+        return f"#### Success 🔥 Yay! This model was successfully exported and a PR was open using your token, here: [{commit_info.pr_url}]({commit_info.pr_url}). If you would like to use the exported model without waiting for the PR to be approved, head to https://huggingface.co/{model_id}/tree/{pr_revision}"
     except Exception as e:
         return f"#### Error: {e}"
 
@@ -151,20 +151,4 @@ with gr.Blocks() as demo:
         outputs=output,
     )
 
-"""
-demo = gr.Interface(
-    title="",
-    description=DESCRIPTION,
-    allow_flagging="never",
-    article="Check out the [🤗 Optimum repoository on GitHub](https://github.com/huggingface/optimum) as well!",
-    inputs=[
-        gr.Text(max_lines=1, label="Hugging Face token"),
-        gr.Text(max_lines=1, label="Model name", placeholder="textattack/distilbert-base-cased-CoLA"),
-        gr.Text(value="auto", max_lines=1, label="Task (can be left blank, will be automatically inferred)")
-    ],
-    outputs=[gr.Markdown(label="output")],
-    fn=onnx_export,
-)
-"""
-
 demo.launch()

onnx_export.py

@@ -1,4 +1,3 @@
-import argparse
 import os
 import shutil
 from tempfile import TemporaryDirectory
@@ -10,9 +9,8 @@ from huggingface_hub import (
 )
 from huggingface_hub.file_download import repo_folder_name
 from optimum.exporters.onnx import main_export
-from optimum.exporters.tasks import TasksManager
 
-SPACES_URL = "https://huggingface.co/spaces/optimum/exporters"
+SPACES_URL = "https://huggingface.co/spaces/onnx/export"
 
 
 def previous_pr(api: "HfApi", model_id: str, pr_title: str) -> Optional["Discussion"]:
@@ -84,7 +82,7 @@ def convert(
         try:
             pr = previous_pr(api, model_id, pr_title)
             if "model.onnx" in filenames and not force:
-                raise Exception(f"Model {model_id} is already converted, skipping..")
+                raise Exception(f"Model {model_id} is already converted, skipping the export.")
             elif pr is not None and not force:
                 url = f"https://huggingface.co/{model_id}/discussions/{pr.num}"
                 new_pr = pr
@@ -111,31 +109,3 @@ The exported ONNX model can be then be consumed by various backends as TensorRT
         finally:
             shutil.rmtree(folder)
         return "0", new_pr
-
-
-if __name__ == "__main__":
-    DESCRIPTION = """
-    Simple utility tool to convert automatically a model on the hub to onnx format.
-    It is PyTorch exclusive for now.
-    It works by downloading the weights (PT), converting them locally, and uploading them back
-    as a PR on the hub.
-    """
-    parser = argparse.ArgumentParser(description=DESCRIPTION)
-    parser.add_argument(
-        "--model_id",
-        type=str,
-        help="The name of the model on the hub to convert. E.g. `gpt2` or `facebook/wav2vec2-base-960h`",
-    )
-    parser.add_argument(
-        "--task",
-        type=str,
-        help="The task the model is performing",
-    )
-    parser.add_argument(
-        "--force",
-        action="store_true",
-        help="Create the PR even if it already exists of if the model was already converted.",
-    )
-    args = parser.parse_args()
-    api = HfApi()
-    convert(api, args.model_id, task=args.task, force=args.force)

requirements.txt

@@ -1,4 +1,4 @@
 torch
 transformers
-git+https://github.com/huggingface/huggingface_hub@main
-git+https://github.com/huggingface/optimum.git#egg=optimum[onnxruntime]
+huggingface_hub
+optimum[onnxruntime,exporters]>=1.9.0