CobaltZvc commited on
Commit
c0c019e
1 Parent(s): 1fff685

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -732,6 +732,7 @@ if Input_type == 'TEXT':
732
  warnings.warn(
733
  "Your request activated the API's safety filters and could not be processed."
734
  "Please modify the prompt and try again.")
 
735
  if artifact.type == generation.ARTIFACT_IMAGE:
736
  img = Image.open(io.BytesIO(artifact.binary))
737
  st.image(img)
 
732
  warnings.warn(
733
  "Your request activated the API's safety filters and could not be processed."
734
  "Please modify the prompt and try again.")
735
+ st.warning("Issue with image generation")
736
  if artifact.type == generation.ARTIFACT_IMAGE:
737
  img = Image.open(io.BytesIO(artifact.binary))
738
  st.image(img)