Yntec commited on
Commit
557d170
·
verified ·
1 Parent(s): fc38712

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -24,7 +24,7 @@ gallery = gr.Gallery(label="Generated image", show_label=False, elem_id="gallery
24
  def infer(prompt):
25
 
26
  #image = pipe(prompt, init_image=init_image)["sample"][0]
27
- images_list = pipe([prompt])
28
  images = []
29
  safe_image = Image.open(r"unsafe.png")
30
  for i, image in enumerate(images_list["images"]):
@@ -38,7 +38,7 @@ def infer(prompt):
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
 
 
24
  def infer(prompt):
25
 
26
  #image = pipe(prompt, init_image=init_image)["sample"][0]
27
+ images_list = pipe([prompt], num_inference_steps=21)
28
  images = []
29
  safe_image = Image.open(r"unsafe.png")
30
  for i, image in enumerate(images_list["images"]):
 
38
  print("Great sylvain ! Everything is working fine !")
39
 
40
  title="YiffyPixar"
41
+ description="It takes about 500 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"
42
 
43
+ gr.Interface(fn=infer, inputs=["text", num_inference_steps], outputs=gallery,title=title,description=description).queue(max_size=10).launch(enable_queue=True)
44