Update app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 import numpy as np
 import random
 import torch
-import torch
+import os
 from diffusers import Transformer2DModel, PixArtSigmaPipeline, AutoencoderKL, DPMSolverMultistepScheduler, DDIMScheduler, EulerAncestralDiscreteScheduler, DPMSolverSDEScheduler
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, QuantoConfig, EetqConfig
 
@@ -15,7 +15,8 @@ MAX_SEED = np.iinfo(np.int32).max
 
 transformer = Transformer2DModel.from_pretrained(
     "alfredplpl/commonart-tmp-3",
-    torch_dtype=weight_dtype
+    torch_dtype=weight_dtype,
+    token=os.environ["TOKEN"]
 )
 
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=weight_dtype)