ptx0 commited on
Commit
70187df
1 Parent(s): 5ff0926

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -8
app.py CHANGED
@@ -10,14 +10,13 @@ pipe.to('cuda')
10
  # Define the image generation function with adjustable parameters and a progress bar
11
  @spaces.GPU
12
  def generate(prompt, guidance_scale, guidance_rescale, num_inference_steps, negative_prompt):
13
- with gr.Progress() as progress:
14
- return pipe(
15
- prompt,
16
- negative_prompt=negative_prompt,
17
- guidance_scale=guidance_scale,
18
- guidance_rescale=guidance_rescale,
19
- num_inference_steps=num_inference_steps
20
- ).images
21
 
22
  # Example prompts to demonstrate the model's capabilities
23
  example_prompts = [
 
10
  # Define the image generation function with adjustable parameters and a progress bar
11
  @spaces.GPU
12
  def generate(prompt, guidance_scale, guidance_rescale, num_inference_steps, negative_prompt):
13
+ return pipe(
14
+ prompt,
15
+ negative_prompt=negative_prompt,
16
+ guidance_scale=guidance_scale,
17
+ guidance_rescale=guidance_rescale,
18
+ num_inference_steps=num_inference_steps
19
+ ).images
 
20
 
21
  # Example prompts to demonstrate the model's capabilities
22
  example_prompts = [