update tok

app.py

@@ -79,19 +79,19 @@ def FlowEditRun(
 
     ):
 
-    if oauth_token is None:
-        raise gr.Error("You must be logged in to use Stable Diffusion 3.0 and FLUX.1 models.")
-    if model_type == 'SD3':
-        try:
-            print(f'token: {oauth_token.token}')
-            huggingface_hub.get_hf_file_metadata(huggingface_hub.hf_hub_url(SD3STRING, 'transformer/diffusion_pytorch_model.safetensors'),
-                                                    token=oauth_token.token)
-            print('Has Access')
-        # except huggingface_hub.utils._errors.GatedRepoError:
-        except huggingface_hub.errors.GatedRepoError:
-            raise gr.Error("You need to accept the license agreement to use Stable Diffusion 3. "
-                            "Visit the <a href='https://huggingface.co/stabilityai/stable-diffusion-3-medium-diffusers'>"
-                            "model page</a> to get access.")
+    # if oauth_token is None:
+    #     raise gr.Error("You must be logged in to use Stable Diffusion 3.0 and FLUX.1 models.")
+    # if model_type == 'SD3':
+    #     try:
+    #         print(f'token: {oauth_token.token}')
+    #         huggingface_hub.get_hf_file_metadata(huggingface_hub.hf_hub_url(SD3STRING, 'transformer/diffusion_pytorch_model.safetensors'),
+    #                                                 token=oauth_token.token)
+    #         print('Has Access')
+    #     # except huggingface_hub.utils._errors.GatedRepoError:
+    #     except huggingface_hub.errors.GatedRepoError:
+    #         raise gr.Error("You need to accept the license agreement to use Stable Diffusion 3. "
+    #                         "Visit the <a href='https://huggingface.co/stabilityai/stable-diffusion-3-medium-diffusers'>"
+    #                         "model page</a> to get access.")
     # elif model_type == 'FLUX':
     #     try:
     #         huggingface_hub.get_hf_file_metadata(huggingface_hub.hf_hub_url(FLUXSTRING, 'flux1-dev.safetensors'),
@@ -119,10 +119,10 @@ def FlowEditRun(
 
         if model_type == 'FLUX':
             # pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.float16) 
-            pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.float16)#, token=os.getenv('PRIV_TOKEN'))
+            pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.float16, token=os.getenv('HF_ACCESS_TOK'))
             loaded_model = 'FLUX'
         elif model_type == 'SD3':
-            pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16)#, token=os.getenv('PRIV_TOKEN'))
+            pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16, token=os.getenv('HF_ACCESS_TOK'))
             loaded_model = 'SD3'
         else:
             raise NotImplementedError(f"Model type {model_type} not implemented")
@@ -239,8 +239,8 @@ with gr.Blocks() as demo:
 
     gr.HTML(intro)
     
-    with gr.Row():
-        gr.LoginButton(value="Login to HF (For SD3 and FLUX access)", variant="primary")
+    # with gr.Row():
+    #     gr.LoginButton(value="Login to HF (For SD3 and FLUX access)", variant="primary")
 
     with gr.Row(equal_height=True):
         image_src = gr.Image(type="filepath", label="Source Image", value="inputs/cat.png",)