Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from transformers import AutoModel
+from diffusers import FluxPipeline
 from huggingface_hub.utils import HfHubHTTPError, RepositoryNotFoundError
 
 def check_model_access(model_id: str, oauth_token: gr.OAuthToken | None):
@@ -25,14 +26,14 @@ def check_model_access(model_id: str, oauth_token: gr.OAuthToken | None):
     try:
         # 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.
-        model = AutoModel.from_pretrained(pretrained_model_name_or_path=model_id, token=oauth_token.token)
+        pipe = FluxPipeline.from_pretrained(pretrained_model_name_or_path=model_id, torch_dtype=torch.bfloat16, token=oauth_token.token)
         
         # 4. If the call succeeds, format a success message
         return f"""
         ### <span style='color: green;'>Access Granted ✅</span>
         Successfully loaded the configuration for **{model_id}**.
         
-        - **Model Type:** `{model}`
+        - **Model Type:** `{pipe}`
         - **Architecture:** 
         - token : {oauth_token.token}
         """