cutycat2000x commited on
Commit
00e3bd5
·
verified ·
1 Parent(s): 19ef2ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -58,7 +58,7 @@ def generate(prompt, negative_prompt, style, use_negative_prompt, num_inference_
58
  negative_prompt = ""
59
  prompt, negative_prompt = apply_style(style, prompt, negative_prompt)
60
 
61
- images = pipe(
62
  prompt=prompt,
63
  negative_prompt=negative_prompt,
64
  width=width,
@@ -68,10 +68,9 @@ def generate(prompt, negative_prompt, style, use_negative_prompt, num_inference_
68
  num_images_per_prompt=num_images_per_prompt,
69
  cross_attention_kwargs={"scale": 0.65},
70
  output_type="pil"
71
- ).images
 
72
 
73
- image_paths = [save_image(img) for img in images]
74
- return image_paths, seed
75
 
76
  examples = [
77
  'a smiling girl with sparkles in her eyes, walking in a garden, in the morning --style anime',
 
58
  negative_prompt = ""
59
  prompt, negative_prompt = apply_style(style, prompt, negative_prompt)
60
 
61
+ result = pipe(
62
  prompt=prompt,
63
  negative_prompt=negative_prompt,
64
  width=width,
 
68
  num_images_per_prompt=num_images_per_prompt,
69
  cross_attention_kwargs={"scale": 0.65},
70
  output_type="pil"
71
+ )
72
+ return result.images, seed
73
 
 
 
74
 
75
  examples = [
76
  'a smiling girl with sparkles in her eyes, walking in a garden, in the morning --style anime',