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 = (480, 480)
     if bg_image:
         img_back=Image.open(bg_image).convert('RGB').resize(img_size)
         comp_img = img_back
@@ -73,4 +73,9 @@ if st.button('Submit'):
         del images
     else:
         st.image(images.images)
-        del images
+        del images
+
+st.warning("If you don't see any image could be the NSFW content checker or a Memory error."
+            "Please, re-run the submission and modify the prompt."
+            "Taking care that, each word in the prompt result in much more token for the model"
+            "which means much more memory usage.", icon="⚠️")