nroggendorff commited on
Commit
9fffbce
1 Parent(s): ae7dc05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ pipeline = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dt
9
 
10
  @spaces.GPU(duration=70)
11
  def generate(prompt, negative_prompt, width, height, sample_steps):
12
- return pipeline(prompt=f"{prompt} anime\nDO NOT INCLUDE {negative_prompt}", width=width, height=height, num_inference_steps=sample_steps, generator=torch.Generator("cpu").manual_seed(42), guidance_scale=7).images[0]
13
 
14
  with gr.Blocks() as interface:
15
  with gr.Column():
 
9
 
10
  @spaces.GPU(duration=70)
11
  def generate(prompt, negative_prompt, width, height, sample_steps):
12
+ return pipeline(prompt=f"{prompt}\nDO NOT INCLUDE {negative_prompt}", width=width, height=height, num_inference_steps=sample_steps, generator=torch.Generator("cpu").manual_seed(42), guidance_scale=7).images[0]
13
 
14
  with gr.Blocks() as interface:
15
  with gr.Column():