|
import gradio as gr |
|
|
|
from huggingface_hub import list_models |
|
|
|
|
|
def list_private_models(oauth_profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken | None) -> str: |
|
if oauth_token is None: |
|
return "Please log in to list private models." |
|
|
|
models = list_models(author=oauth_profile.username, token=oauth_token.token) |
|
|
|
return f"Private models: {', '.join(model.id for model in models if model.private)}" |
|
|
|
with gr.Blocks() as demo: |
|
gr.LoginButton() |
|
gr.LogoutButton() |
|
m1 = gr.Markdown() |
|
demo.load(list_private_models, inputs=None, outputs=m1) |
|
|
|
demo.launch() |
|
|