Update app.py

app.py

@@ -1,13 +1,11 @@
 import gradio as gr
-from huggingface_hub import HfApi
+from transformers import AutoConfig
 from huggingface_hub.utils import HfHubHTTPError, RepositoryNotFoundError
 
-# Initialize the Hugging Face Hub API client
-api = HfApi()
-
 def check_model_access(model_id: str, oauth_token: gr.OAuthToken | None):
     """
-    Checks if the logged-in user has access to a given model on the Hub.
+    Checks if the logged-in user can load the configuration for a given model on the Hub.
+    Loading the config proves file-level access to the repository.
 
     Args:
         model_id: The ID of the model to check (e.g., "meta-llama/Llama-2-7b-chat-hf").
@@ -25,18 +23,17 @@ def check_model_access(model_id: str, oauth_token: gr.OAuthToken | None):
         return "### <span style='color: orange;'>Input Missing 🟡</span>\nPlease enter a model ID to check."
 
     try:
-        # 3. The core of the test: try to get model info using the user's token
-        # This single API call is enough to verify read access.
-        model_info = api.model_info(repo_id=model_id, token=oauth_token.token)
+        # 3. The core test: try to load the model's config using the user's token.
+        # This will fail if the user doesn't have access to the gated repo's files.
+        config = AutoConfig.from_pretrained(pretrained_model_name_or_path=model_id, token=oauth_token.token)
         
         # 4. If the call succeeds, format a success message
         return f"""
         ### <span style='color: green;'>Access Granted ✅</span>
-        You have access to **{model_id}**.
+        Successfully loaded the configuration for **{model_id}**.
         
-        - **Author:** {model_info.author}
-        - **Last Modified:** {model_info.last_modified}
-        - **Gated:** {'Yes' if model_info.gated else 'No'}
+        - **Model Type:** `{config.model_type}`
+        - **Architecture:** `{config.architectures[0] if config.architectures else 'N/A'}`
         """
     
     except RepositoryNotFoundError:
@@ -45,24 +42,23 @@ def check_model_access(model_id: str, oauth_token: gr.OAuthToken | None):
         
     except HfHubHTTPError as e:
         # 6. Handle HTTP errors, which typically indicate permission issues for gated models
-        if e.response.status_code == 401 or e.response.status_code == 403:
+        if e.response.status_code in [401, 403]:
             return f"""
             ### <span style='color: red;'>Access Denied 🔴</span>
-            You do not have access to **{model_id}**.
+            You do not have permission to download files from **{model_id}**.
             
             - Please ensure you have accepted the terms and conditions on the model's page.
-            - This might be a private model you don't have permission for.
+            - This might be a private model you don't have access to.
             - **Status Code:** {e.response.status_code}
             """
         else:
             return f"### <span style='color: red;'>An Error Occurred 🔴</span>\n**Details:** {str(e)}"
 
-# --- Gradio Interface ---
+# --- Gradio Interface (No changes needed here) ---
 with gr.Blocks(css="h1 { text-align: center; }") as demo:
     gr.Markdown("# Gated Model Access Tester")
-    gr.Markdown("Log in with your Hugging Face account and enter a model ID to see if you have access.")
+    gr.Markdown("Log in with your Hugging Face account and enter a model ID. This will attempt to load the model's `config.json` file to verify access.")
     
-    # The LoginButton is the key to enabling OAuth
     gr.LoginButton()
 
     with gr.Row():
@@ -75,8 +71,6 @@ with gr.Blocks(css="h1 { text-align: center; }") as demo:
 
     result_display = gr.Markdown("### Result will be displayed here.")
 
-    # When the button is clicked, call the check_model_access function.
-    # Gradio automatically finds the oauth_token from the session and passes it.
     check_button.click(
         fn=check_model_access,
         inputs=[model_id_input],