Yntec commited on
Commit
6928ae2
·
verified ·
1 Parent(s): 2cf8757

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -16,10 +16,10 @@ YOUR_TOKEN=MY_SECRET_TOKEN
16
 
17
  device="cpu"
18
 
19
- pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", use_auth_token=YOUR_TOKEN)
20
  pipe.to(device)
21
 
22
- gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
23
 
24
  def infer(prompt):
25
 
@@ -37,8 +37,8 @@ def infer(prompt):
37
 
38
  print("Great sylvain ! Everything is working fine !")
39
 
40
- title="Stable Diffusion CPU"
41
- description="Stable Diffusion example using CPU and HF token. <br />Warning: Slow process... ~5/10 min inference time. <b>NSFW filter enabled.</b>"
42
 
43
  gr.Interface(fn=infer, inputs="text", outputs=gallery,title=title,description=description).queue(max_size=10).launch(enable_queue=True)
44
 
 
16
 
17
  device="cpu"
18
 
19
+ pipe = StableDiffusionPipeline.from_pretrained("Yntec/YiffyPixar", use_auth_token=YOUR_TOKEN)
20
  pipe.to(device)
21
 
22
+ gallery = gr.Gallery(label="Generated image", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
23
 
24
  def infer(prompt):
25
 
 
37
 
38
  print("Great sylvain ! Everything is working fine !")
39
 
40
+ title="YiffyPixar"
41
+ description="It takes about 1000 seconds to generate an image per user. If you like this model, you can like it at its original page! https://huggingface.co/Yntec/YiffyPixar/tree/main"
42
 
43
  gr.Interface(fn=infer, inputs="text", outputs=gallery,title=title,description=description).queue(max_size=10).launch(enable_queue=True)
44