Update app.py

app.py

@@ -52,6 +52,8 @@ GENERATE_IMAGE_TEXT = "Generate NFT by selecting existing model based on OpenSea
 INTERPOLATION_TEXT = "Generate interpolation between two NFTs by selecting existing model based on OpenSea collection. You can create new model or imporve existing in few clicks — check [project repository](https://github.com/AlekseyKorshuk/huggingnft)."
 COLLECTION2COLLECTION_TEXT = "Generate first NFT with existing model and transform it to another collection by selecting existing model based on OpenSea collections. You can create new model or imporve existing in few clicks — check [project repository](https://github.com/AlekseyKorshuk/huggingnft)."
 
+TRAIN_TEXT = "If you think that the results of the model are not good enough and they can be improved, you can train the model more in a few clicks. If you notice that the model is overtrained, then you can easily return to the best version. Check [project repository](https://github.com/AlekseyKorshuk/huggingnft) to know more about that."
+
 STOPWORDS = ["-old"]
 COLLECTION2COLLECTION_KEYS = ["__2__"]
 
@@ -162,13 +164,15 @@ if choose == "Generate image":
         with st.spinner(text=f"Downloading selected model..."):
             model = load_lightweight_model(f"huggingnft/{model_name}")
         with st.spinner(text=f"Generating..."):
-            st.image(
-                model.generate_app(
+            image = model.generate_app(
                     num=timestamped_filename(),
                     nrow=nrows,
                     checkpoint=-1,
                     types=generation_type
                 )[0]
+        st.markdown(TRAIN_TEXT)
+        st.image(
+                image
             )
 
 if choose == "Interpolation":
@@ -206,8 +210,10 @@ if choose == "Interpolation":
             progress_bar=my_bar
         )
         my_bar.empty()
-        with st.spinner(text=f"Uploading result..."):
-            st.image(result)
+        st.markdown(TRAIN_TEXT)
+        st.image(
+                result
+            )
 
 if choose == "Collection2Collection":
     st.title(choose)
@@ -280,5 +286,6 @@ if choose == "Collection2Collection":
             results.append(
                 get_concat_h(out_transform(make_grid(out_punk, nrow=1, normalize=True)), out_transform(make_grid(out_ape, nrow=1, normalize=True)))
             )
+        st.markdown(TRAIN_TEXT)
         for result in results:
             st.image(result)