Update app.py

app.py

@@ -6,6 +6,7 @@ is_spaces = True if os.environ.get("SPACE_ID") else False
 if is_spaces:
     subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
     import spaces
+    from huggingface_hub import whoami
     
     
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
@@ -230,8 +231,9 @@ h3{margin-top: 0}
 .tabitem{border: 0px}
 """
 
-def swap_visibilty(profile: Union[gr.OAuthProfile, None]):
-    print(profile)
+def swap_visibilty(profile: Union[gr.OAuthProfile, None], oauth_token: Union[gr.OAuthToken, None]):
+    user = whoami(oauth_token.token)
+    print(user)
     if is_spaces:
         if profile is None:
             return gr.update(elem_classes=["main_ui_logged_out"])