add token

app.py

@@ -1,6 +1,9 @@
+import os
+
 import gradio as gr
 from huggingface_hub import whoami
 
+
 log_file = "out-file.txt"
 
 
@@ -24,19 +27,16 @@ def run():
         yield content
 
 
-
-
 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:
+def set_token(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(oauth_token.token)["orgs"]]
-    return f"You belong to {', '.join(org_names)}."
+        return "Please log in."
+    os.environ["HF_TOKEN"] = oauth_token.token
 
 
 with gr.Blocks() as demo:
@@ -44,7 +44,7 @@ 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(set_token, inputs=None, outputs=m2)
 
     button = gr.Button("Run pipeline")
     output = gr.Code(label="Terminal output", interactive=False)