mfranzon commited on
Commit
aaa1cf1
1 Parent(s): 16824b4

fix main utils

Browse files
Files changed (2) hide show
  1. main.py +2 -4
  2. utils.py +1 -1
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
- num_inference_steps=50)
26
  return images
22
  init_image=init_image,
23
  strength=strength,
24
  guidance_scale=7.5,
25
+ )
26
  return images