Update app.py

app.py

@@ -86,8 +86,12 @@ def guess_base_model(ft_model_id):
 
 
 def process_model(ft_model_id: str, base_model_id: str, rank: str, private_repo, oauth_token: gr.OAuthToken | None):
-    if oauth_token is None or oauth_token.token is None:
+    # validate the oauth token
+    try:
+        whoami(oauth_token.token)
+    except Exception as e:
         raise gr.Error("You must be logged in")
+
     model_name = ft_model_id.split('/')[-1]
 
     # validate the oauth token