| from __future__ import annotations |
|
|
| import gradio as gr |
| from huggingface_hub import whoami |
|
|
| def hello(profile: gr.OAuthProfile | None) -> str: |
| if profile is None: |
| return "I don't know you." |
| return f"Hello {profile.name}" |
|
|
| def list_organizations(oauth_token: gr.OAuthToken | None) -> str: |
| if oauth_token is None: |
| return "Please deploy this on Spaces and log in to list organizations." |
| org_names = [org["name"] for org in whoami(oauth_token.token)["orgs"]] |
| return f"You belong to {', '.join(org_names)}." |
|
|
| with gr.Blocks() as demo: |
| gr.LoginButton() |
| m1 = gr.Markdown() |
| m2 = gr.Markdown() |
| demo.load(hello, inputs=None, outputs=m1) |
| demo.load(list_organizations, inputs=None, outputs=m2) |
|
|
| demo.launch() |
