Update app.py

app.py

@@ -4,16 +4,18 @@ import gradio as gr
 import torch
 from PIL import Image
 import random
+import os
 
 model_id = 'aipicasso/picasso-diffusion-1-0'
+token=os.environ.get("ACCESS_TOKEN")
 
-scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
-feature_extractor = CLIPFeatureExtractor.from_pretrained(model_id)
+scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler", use_auth_token=token)
+feature_extractor = CLIPFeatureExtractor.from_pretrained(model_id, use_auth_token=token)
 
 pipe = StableDiffusionPipeline.from_pretrained(
   model_id,
   torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
-  scheduler=scheduler)
+  scheduler=scheduler, use_auth_token=token)
 pipe.enable_xformers_memory_efficient_attention()
 
 pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
@@ -22,7 +24,7 @@ pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
   scheduler=scheduler,
   requires_safety_checker=False,
   safety_checker=None,
-  feature_extractor=feature_extractor
+  feature_extractor=feature_extractor, use_auth_token=token
 )
 pipe_i2i.enable_xformers_memory_efficient_attention()