diff --git a/README.md b/README.md index 4d49e448..d5dfa954 100644 --- a/README.md +++ b/README.md @@ -6,9 +6,7 @@ Application to generate your training scripts with [PyTorch-Ignite](https://github.com/pytorch/ignite). -
-Here should be an image and link to deployed application -
+[![Code-Generator](https://user-images.githubusercontent.com/32727188/115562045-214aff00-a2dc-11eb-9de5-bec9fa7b94d2.png)][streamlit-url] ## Contribution diff --git a/app/streamlit_app.py b/app/streamlit_app.py index 412f90de..6541ddd1 100644 --- a/app/streamlit_app.py +++ b/app/streamlit_app.py @@ -84,6 +84,10 @@ class App: def __init__(self): st.set_page_config(page_title=self.page_title, page_icon=self.page_icon) + st.info( + "Code-Generator v0.2.0 released with new UI, templates, and bug fixes.\ + \nTry now at: https://code-generator.pytorch-ignite.ai/" + ) st.write(self.description, unsafe_allow_html=True) self.codegen = CodeGenerator()