oauth

README.md

@@ -4,9 +4,13 @@ emoji: 📈
 colorFrom: yellow
 colorTo: purple
 sdk: gradio
-sdk_version: 4.26.0
+sdk_version: 4.27.0
 app_file: app.py
 pinned: false
+hf_oauth: true
+hf_oauth_scopes:
+  - read-repos
+  - write-repos
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -45,7 +45,9 @@ you've downloaded from your Kaggle user account.
 """
 
 
-def import_model(kaggle_model: str, repo_name: str, token: str) -> Iterable[List[Log]]:
+def import_model(
+    kaggle_model: str, repo_name: str, token: gr.OAuthToken | None
+) -> Iterable[List[Log]]:
     runner = LogsViewRunner()
 
     if not kaggle_model:
@@ -54,10 +56,9 @@ def import_model(kaggle_model: str, repo_name: str, token: str) -> Iterable[List
     if not repo_name:
         repo_name = kaggle_model.split("/")[-1]
     if not token:
-        yield runner.log("HF Write Token is required.", level="ERROR")
-        raise gr.Error("HF Write Token is required.")
-    api = HfApi(token=token)
-
+        yield runner.log("You must sign in with HF before proceeding.", level="ERROR")
+        raise gr.Error("Authentication is required.")
+    api = HfApi(token=token.token)
 
     yield runner.log(f"Creating HF repo {repo_name}")
     repo_url = api.create_repo(repo_name, exist_ok=True)
@@ -102,32 +103,22 @@ with gr.Blocks() as demo:
     gr.Markdown(MARKDOWN_DESCRIPTION)
 
     with gr.Row():
-        with gr.Column():
-            kaggle_model = gr.Textbox(
-                lines=1,
-                label="Kaggle Model*",
-                placeholder="keras/codegemma/keras/code_gemma_7b_en",
-            )
-            repo_name = gr.Textbox(
-                lines=1,
-                label="Repo name",
-                placeholder="Optional. Will infer from Kaggle Model if empty.",
-            )
-        with gr.Column():
-            token = gr.Textbox(
-                lines=1,
-                label="HF Write Token*",
-                info="https://hf.co/settings/token",
-                type="password",
-                placeholder="hf_***",
-            )
+        kaggle_model = gr.Textbox(
+            lines=1,
+            label="Kaggle Model*",
+            placeholder="keras/codegemma/keras/code_gemma_7b_en",
+        )
+        repo_name = gr.Textbox(
+            lines=1,
+            label="Repo name",
+            placeholder="Optional. Will infer from Kaggle Model if empty.",
+        )
+        gr.LoginButton()
 
     button = gr.Button("Import", variant="primary")
     logs = LogsView(label="Terminal output")
 
-    button.click(
-        fn=import_model, inputs=[kaggle_model, repo_name, token], outputs=[logs]
-    )
+    button.click(fn=import_model, inputs=[kaggle_model, repo_name], outputs=[logs])
 
 
 demo.queue(default_concurrency_limit=1).launch()