Staticaliza commited on
Commit
c7f68b4
·
verified ·
1 Parent(s): e9947e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -43,10 +43,10 @@ footer {
43
  '''
44
 
45
  # Functions
46
- def save_image(img):
47
- unique_name = str(uuid.uuid4()) + ".png"
48
- img.save(unique_name)
49
- return unique_name
50
 
51
  def get_seed(seed):
52
  seed = seed.strip()
@@ -71,14 +71,14 @@ def generate(input=DEFAULT_INPUT, negative_input=DEFAULT_NEGATIVE_INPUT, height=
71
  "num_inference_steps": steps,
72
  "guidance_scale": guidance,
73
  "num_images_per_prompt": number,
74
- "cross_attention_kwargs": {"scale": 0.65},
75
  "generator": torch.Generator().manual_seed(seed),
76
  "use_resolution_binning": True,
77
  "output_type":"pil",
78
  }
79
 
80
  images = model(**parameters).images
81
- image_paths = [save_image(img) for img in images]
82
  print(image_paths)
83
  return image_paths
84
 
 
43
  '''
44
 
45
  # Functions
46
+ def save_image(img, seed):
47
+ name = f"{seed}-{uuid.uuid4()}.png"
48
+ img.save(name)
49
+ return name
50
 
51
  def get_seed(seed):
52
  seed = seed.strip()
 
71
  "num_inference_steps": steps,
72
  "guidance_scale": guidance,
73
  "num_images_per_prompt": number,
74
+ "cross_attention_kwargs": {"scale": 0.01},
75
  "generator": torch.Generator().manual_seed(seed),
76
  "use_resolution_binning": True,
77
  "output_type":"pil",
78
  }
79
 
80
  images = model(**parameters).images
81
+ image_paths = [save_image(img, seed) for img in images]
82
  print(image_paths)
83
  return image_paths
84