Update app.py

app.py

@@ -28,6 +28,17 @@ USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE") == "1"
 ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
 OUTPUT_DIR = os.getenv("OUTPUT_DIR", "./outputs")
 
+from diffusers.utils import make_image_grid
+
+ACCESS_TOKEN = os.environ["ACCESS_TOKEN"]
+login(token=ACCESS_TOKEN)
+pipeline = DiffusionPipeline.from_pretrained(
+    "Tasty-Rice/Magic_on_paper/blob/main/Magic_on_paper-SDXL-v3.safetensors",
+    trust_remote_code=True,
+    torch_dtype=torch.float16,
+    use_auth_token=ACCESS_TOKEN
+)
+
 MODEL = os.getenv(
     "MODEL",
     "https://huggingface.co/Tasty-Rice/Magic_on_paper/blob/main/Magic_on_paper-SDXL-v3.safetensors",
@@ -37,7 +48,7 @@ torch.backends.cudnn.deterministic = True
 torch.backends.cudnn.benchmark = False
 
 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
-MODEL.to(device)
+load_pipeline.to(device)
 
 def load_pipeline(model_name):
     vae = AutoencoderKL.from_pretrained(