mkshing commited on
Commit
e271893
1 Parent(s): b54e7bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import os
2
  import random
 
3
  import gradio as gr
4
  import torch
5
  from diffusers import (
@@ -15,6 +16,7 @@ from diffusers import (
15
  from diffusers.utils import make_image_grid
16
 
17
  ACCESS_TOKEN = os.environ["ACCESS_TOKEN"]
 
18
  pipeline = DiffusionPipeline.from_pretrained(
19
  "stabilityai/japanese-stable-diffusion-xl",
20
  trust_remote_code=True,
 
1
  import os
2
  import random
3
+ from huggingface_hub import login
4
  import gradio as gr
5
  import torch
6
  from diffusers import (
 
16
  from diffusers.utils import make_image_grid
17
 
18
  ACCESS_TOKEN = os.environ["ACCESS_TOKEN"]
19
+ login(token=ACCESS_TOKEN)
20
  pipeline = DiffusionPipeline.from_pretrained(
21
  "stabilityai/japanese-stable-diffusion-xl",
22
  trust_remote_code=True,