Hev832 commited on
Commit
274e2b7
·
verified ·
1 Parent(s): e440e66

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,7 +4,8 @@ from diffusers import DiffusionPipeline
4
  # Load the pipeline
5
  pipeline = DiffusionPipeline.from_pretrained("John6666/t-ponynai3-v6-sdxl")
6
 
7
- def generate_image(prompt, negative_prompt):
 
8
  # Generate image using the pipeline with both prompt and negative prompt
9
  image = pipeline(prompt, negative_prompt=negative_prompt).images[0]
10
  return image
 
4
  # Load the pipeline
5
  pipeline = DiffusionPipeline.from_pretrained("John6666/t-ponynai3-v6-sdxl")
6
 
7
+ def generate_image(prompt, negative_prompt,progress=gr.Progress()):
8
+ progress(0, desc="Starting Pipeline, Please Wait...")
9
  # Generate image using the pipeline with both prompt and negative prompt
10
  image = pipeline(prompt, negative_prompt=negative_prompt).images[0]
11
  return image