oauth 3

app.py

@@ -3,6 +3,7 @@ import torch
 from diffusers import FluxPipeline, StableDiffusion3Pipeline
 from PIL import Image
 from typing import Optional
+import os
 
 import random
 import numpy as np
@@ -93,10 +94,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=oauth_token.token)
+            pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.float16, token=os.getenv('PRIV_TOKEN'))
             loaded_model = 'FLUX'
         elif model_type == 'SD3':
-            pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16, token=oauth_token.token)
+            pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16, token=os.getenv('PRIV_TOKEN'))
             loaded_model = 'SD3'
         else:
             raise NotImplementedError(f"Model type {model_type} not implemented")