Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -11,6 +11,8 @@ MY_SECRET_TOKEN=os.environ.get('HF_TOKEN_SD')
|
|
11 |
#from diffusers import StableDiffusionPipeline
|
12 |
from diffusers import StableDiffusionImg2ImgPipeline
|
13 |
|
|
|
|
|
14 |
print("hello")
|
15 |
|
16 |
YOUR_TOKEN=MY_SECRET_TOKEN
|
@@ -18,6 +20,7 @@ YOUR_TOKEN=MY_SECRET_TOKEN
|
|
18 |
device="cpu"
|
19 |
|
20 |
img_pipe = StableDiffusionImg2ImgPipeline.from_pretrained("AkiKagura/mkgen-diffusion", duse_auth_token=YOUR_TOKEN)
|
|
|
21 |
img_pipe.to(device)
|
22 |
|
23 |
source_img = gr.Image(source="upload", type="filepath", label="init_img | 512*512 px")
|
|
|
11 |
#from diffusers import StableDiffusionPipeline
|
12 |
from diffusers import StableDiffusionImg2ImgPipeline
|
13 |
|
14 |
+
def empty_checker(images, **kwargs): return images, False
|
15 |
+
|
16 |
print("hello")
|
17 |
|
18 |
YOUR_TOKEN=MY_SECRET_TOKEN
|
|
|
20 |
device="cpu"
|
21 |
|
22 |
img_pipe = StableDiffusionImg2ImgPipeline.from_pretrained("AkiKagura/mkgen-diffusion", duse_auth_token=YOUR_TOKEN)
|
23 |
+
img_pipe.safety_checker = empty_checker
|
24 |
img_pipe.to(device)
|
25 |
|
26 |
source_img = gr.Image(source="upload", type="filepath", label="init_img | 512*512 px")
|