Spaces:
Running
Running
Update app.py
Browse files
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)
|