update

README.md

@@ -4,7 +4,7 @@ emoji: 🏆
 colorFrom: pink
 colorTo: pink
 sdk: gradio
-sdk_version: 4.9.0
+sdk_version: 4.14.0
 hf_oauth: true
 hf_oauth_scopes:
 - email

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from huggingface_hub import whoami
+from huggingface_hub import list_models
 
 
 def hello(profile: gr.OAuthProfile | None) -> str:
@@ -7,11 +7,14 @@ def hello(profile: gr.OAuthProfile | None) -> str:
         return "I don't know you."
     return f"Hello {profile.name}"
 
-def list_organizations(oauth_token: gr.OAuthToken | None) -> str:
+def list_private_models(profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken | None) -> str:
     if oauth_token is None:
-        return "Please log in to list organizations."
-    org_names = [org["name"] for org in whoami(token=oauth_token.token)["orgs"]]
-    return f"You belong to {', '.join(org_names)}."
+        return "Please log in to list private models."
+    models = [
+        f"{model.id} ({'private' if model.private else 'public'})"
+        for model in list_models(author=profile.username, token=oauth_token.token)
+    ]
+    return "Models:\n\n" + "\n - ".join(models) + "."
 
 with gr.Blocks() as demo:
     gr.Markdown(
@@ -22,6 +25,6 @@ with gr.Blocks() as demo:
     m1 = gr.Markdown()
     m2 = gr.Markdown()
     demo.load(hello, inputs=None, outputs=m1)
-    demo.load(list_organizations, inputs=None, outputs=m2)
+    demo.load(list_private_models, inputs=None, outputs=m2)
 
 demo.launch()