Spaces:
Runtime error
Runtime error
fix main utils
Browse files
main.py
CHANGED
@@ -70,12 +70,10 @@ if st.button('Submit'):
|
|
70 |
strength=strength)
|
71 |
if images["nsfw_content_detected"]:
|
72 |
st.write("NSFW content detected, retry with another prompt or image")
|
73 |
-
del images
|
74 |
else:
|
75 |
st.image(images.images)
|
76 |
-
del images
|
77 |
|
78 |
-
st.warning("If you don't see any image could be the NSFW content checker or a Memory error."
|
79 |
-
"Please, re-run the submission and modify the prompt."
|
80 |
"Taking care that, each word in the prompt result in much more token for the model"
|
81 |
"which means much more memory usage.")
|
70 |
strength=strength)
|
71 |
if images["nsfw_content_detected"]:
|
72 |
st.write("NSFW content detected, retry with another prompt or image")
|
|
|
73 |
else:
|
74 |
st.image(images.images)
|
|
|
75 |
|
76 |
+
st.warning("If you don't see any image could be the NSFW content checker or a Memory error. "
|
77 |
+
"Please, re-run the submission and modify the prompt. "
|
78 |
"Taking care that, each word in the prompt result in much more token for the model"
|
79 |
"which means much more memory usage.")
|
utils.py
CHANGED
@@ -22,5 +22,5 @@ def pipe_image(prompt,
|
|
22 |
init_image=init_image,
|
23 |
strength=strength,
|
24 |
guidance_scale=7.5,
|
25 |
-
|
26 |
return images
|
22 |
init_image=init_image,
|
23 |
strength=strength,
|
24 |
guidance_scale=7.5,
|
25 |
+
)
|
26 |
return images
|