RanM commited on
Commit
bf5575b
·
verified ·
1 Parent(s): e4a8042

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -15,6 +15,7 @@ def generate_image(prompt, prompt_name):
15
  try:
16
  print(f"Generating image for {prompt_name}")
17
  output = model(prompt=prompt, num_inference_steps=1, guidance_scale=0.0)
 
18
  if output and hasattr(output, 'images') and output.images:
19
  print(f"Image generated for {prompt_name}")
20
  image = output.images[0]
 
15
  try:
16
  print(f"Generating image for {prompt_name}")
17
  output = model(prompt=prompt, num_inference_steps=1, guidance_scale=0.0)
18
+
19
  if output and hasattr(output, 'images') and output.images:
20
  print(f"Image generated for {prompt_name}")
21
  image = output.images[0]