add gradio authentication token

app.py

@@ -19,7 +19,12 @@ if HF_TOKEN:
     repo = Repository(local_dir=DATA_DIR, clone_from=DATASET_REPO_URL, token=HF_TOKEN)
 
 
-def export(model_id: str, token: str, task: str = "auto") -> str:
+def export(model_id: str, task: str, oauth_token: gr.OAuthToken) -> str:
+
+    if oauth_token.token is None:
+        raise ValueError("You must be logged in to use this space")
+    token = oauth_token.token
+
     if model_id == "" or token == "":
         return """
         ### Invalid input 🐞
@@ -91,40 +96,38 @@ After the model conversion, we will open a PR against the source repo to add the
 
 To export your model you need:
 - A Model ID from the Hub
-- A read-access token from [https://huggingface.co/settings/tokens](https://huggingface.co/settings/tokens).
 
 That's it ! 🔥
 """
 
+
+
+model_id = HuggingfaceHubSearch(
+    label="Hub Model ID",
+    placeholder="Search for model id on the hub",
+    search_type="model",
+)
+task = gr.Textbox(
+    value="auto",
+    label="Task : can be left to auto, will be automatically inferred",
+)
 interface = gr.Interface(
     fn=export,
     inputs=[
-        HuggingfaceHubSearch(
-            label="Hub Model ID",
-            placeholder="Search for model id on Huggingface",
-            search_type="model",
-        ),
-        gr.Textbox(
-            max_lines=1,
-            label="Hugging Face token",
-        ),
-        gr.Textbox(
-            label="Task",
-            info="Can be left to auto, will be automatically inferred",
-            placeholder="auto",
-            max_lines=1,
-        ),
+        model_id,
+        task,
     ],
     outputs=[
         gr.Markdown(label="output"),
     ],
-    submit_btn=gr.Button("Export"),
     title=TITLE,
     description=DESCRIPTION,
+    api_name=False,
 )
+
 with gr.Blocks() as demo:
-    # gr.HTML(TTILE_IMAGE)
-    # gr.HTML(TITLE)
+    gr.Markdown("You must be logged in to use this space")
+    gr.LoginButton(min_width=250)
     interface.render()
 
 demo.launch()

export.py

@@ -7,7 +7,9 @@ from typing import List, Optional, Tuple
 import torch
 
 from huggingface_hub import (
+    CommitInfo,
     CommitOperationAdd,
+    Discussion,
     HfApi,
     get_repo_discussions,
     hf_hub_download,