Unconditionally add token, format tweaks.

app.py

@@ -48,28 +48,31 @@ push_mapping = {
 }
 push_labels = list(push_mapping.keys())
 
-def error_str(error, title="Error", model=None, task=None, framework=None, compute_units=None, precision=None, tolerance=None, destination=None):
+def error_str(error, title="Error", model=None, task=None, framework=None, compute_units=None, precision=None, tolerance=None, destination=None, open_discussion=True):
     if not error: return ""
 
-    issue_title = urllib.parse.quote(f"Error converting {model}")
-    issue_description = urllib.parse.quote(f"""Conversion Settings:
-
-        Model: {model}
-        Task: {task}
-        Framework: {framework}
-        Compute Units: {compute_units}
-        Precision: {precision}
-        Tolerance: {tolerance}
-        Push to: {destination}
-
-        Error: {error}
-        """)
-    issue_url = f"https://huggingface.co/spaces/pcuenq/transformers-to-coreml/discussions/new?title={issue_title}&description={issue_description}"
+    discussion_text = ""
+    if open_discussion:
+        issue_title = urllib.parse.quote(f"Error converting {model}")
+        issue_description = urllib.parse.quote(f"""Conversion Settings:
+
+            Model: {model}
+            Task: {task}
+            Framework: {framework}
+            Compute Units: {compute_units}
+            Precision: {precision}
+            Tolerance: {tolerance}
+            Push to: {destination}
+
+            Error: {error}
+            """)
+        issue_url = f"https://huggingface.co/spaces/pcuenq/transformers-to-coreml/discussions/new?title={issue_title}&description={issue_description}"
+        discussion_text = f"It could be that the model is not yet compatible with the Core ML exporter. Please, open a discussion on the [Hugging Face Hub]({issue_url}) to report this issue."
     return f"""
         #### {title}
         {error}
 
-        It could be that the model is not yet compatible with the Core ML exporter. Please, open a discussion on the [Hugging Face Hub]({issue_url}) to report this issue.
+        {discussion_text}
         """
 
 def url_to_model_id(model_id_str):
@@ -194,7 +197,9 @@ def convert(model_id, task,
     push_destination = push_mapping[push_destination]
     if push_destination == "pr":
         destination_model = model_id
-        token = None
+
+    if token is None or token == "":
+        return error_str("Please provide a token to push to the Hub.", open_discussion=False)
 
     # TODO: support legacy format
     base = Path("exported")/model_id
@@ -284,6 +289,7 @@ with gr.Blocks() as demo:
                 max_lines=1,
                 label="Model name or URL, such as apple/mobilevit-small",
                 placeholder="pcuenq/distilbert-base-uncased",
+                value="pcuenq/distilbert-base-uncased",
             )
             btn_get_tasks = gr.Button("Load")
         with gr.Column(scale=3):
@@ -313,26 +319,26 @@ with gr.Blocks() as demo:
                     value=tolerance_labels[0],
                 )
 
-                radio_push = gr.Radio(
-                    label="Destination Model",
-                    choices=push_labels,
-                    value=push_labels[0],
-                )
-                with gr.Row(visible=False) as row_destination:
+                with gr.Group():
+                    text_token = gr.Textbox(label="Hugging Face Token", placeholder="hf_xxxx", value="")
+                    radio_push = gr.Radio(
+                        label="Destination Model",
+                        choices=push_labels,
+                        value=push_labels[0],
+                    )
                     # TODO: public/private
-                    text_destination = gr.Textbox(label="Destination model name", value="")
-                    text_token = gr.Textbox(label="Token (write permissions)", value="")
+                    text_destination = gr.Textbox(visible=False, label="Destination model name", value="")
 
-                btn_convert = gr.Button("Convert")
+                btn_convert = gr.Button("Convert & Push")
                 gr.Markdown("Conversion will take a few minutes.")
 
 
     error_output = gr.Markdown(label="Output")
 
     # Clear output
-    btn_get_tasks.click(lambda x: gr.update(value=''), [], [error_output])
-    btn_convert.click(lambda x: gr.update(value=''), [], [error_output])
-    input_model.submit(lambda x: gr.update(value=''), [],[error_output])
+    btn_get_tasks.click(lambda _: gr.update(value=''), None, error_output)
+    input_model.submit(lambda _: gr.update(value=''), None, error_output)
+    btn_convert.click(lambda _: gr.update(value=''), None, error_output)
 
     input_model.submit(
         fn=on_model_change,
@@ -359,9 +365,9 @@ with gr.Blocks() as demo:
     radio_push.change(
         lambda x: gr.update(visible=x == "Create a new repo"),
         inputs=radio_push,
-        outputs=row_destination,
+        outputs=text_destination,
         queue=False,
-        scroll_to_output=True
+        scroll_to_output=False
     )
 
     gr.HTML("""