fix main

main.py

@@ -46,7 +46,7 @@ prompt = st.text_input(_('Prompt to generate your cool image'),
                placeholder=_('write you text here to improve the image with your favourite style'))
 
 if st.button('Submit'):
-    img_size = (512, 512)
+    img_size = (320, 240)
     if bg_image:
         img_back=Image.open(bg_image).convert('RGB').resize(img_size)
         comp_img = img_back
@@ -70,5 +70,7 @@ if st.button('Submit'):
                        strength=strength)
     if images["nsfw_content_detected"]:
         st.write("NSFW content detected, retry with another prompt or image")
+        del images
     else:
-        st.image(images.images)
+        st.image(images.images)
+        del images