RanM commited on
Commit
194be56
·
verified ·
1 Parent(s): 6292767

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -36,6 +36,7 @@ def generate_image(prompt, prompt_name):
36
  image_bytes = buffered.getvalue()
37
  return image_bytes
38
  else:
 
39
  raise ValueError(f"No images found in model output for {prompt_name}")
40
  except Exception as e:
41
  print(f"An error occurred while generating image for {prompt_name}: {e}")
 
36
  image_bytes = buffered.getvalue()
37
  return image_bytes
38
  else:
39
+ print(f"No images found in model output for {prompt_name}")
40
  raise ValueError(f"No images found in model output for {prompt_name}")
41
  except Exception as e:
42
  print(f"An error occurred while generating image for {prompt_name}: {e}")