add documentation

app.py

@@ -271,6 +271,28 @@ def main():
     if not ('input_image' in st.session_state and st.session_state['input_image'] is not None):
         print("Image not present")
         st.success("Upload an image to start")
+        st.write("Welcome to the interior design controlnet demo! "
+                 "You can start by uploading a picture of your room, after which you will see "
+                 "a good variety of options to edit your current room to generate the room of your dreams! "
+                 "You can choose between inpainting, segmentation conditioning and re-generating objects, which "
+                 "use our custom trained controlnet model."
+                 )
+        st.write("### About the dataset")
+        st.write("To make this demo as good as possible, our team spend a lot of time training a custom model. "
+                 "We used the LAION5B dataset to build our custom dataset, which contains 130k images of 15 types of rooms "
+                 "in almost 30 design styles. After fetching all these images, we started adding metadata such as "
+                 "captions (from the BLIP captioning model) and segmentation maps (from the HuggingFace UperNetForSemanticSegmentation model). "
+                 )
+        st.write("### About the model")
+        st.write(
+            "These were then used to train the controlnet model to generate quality interior design images by using "
+            "the segmentation maps and prompts as conditioning information for the model. "
+            "By training on segmentation maps, the enduser has a very finegrained control over which objects they "
+            "want to place in their room. "
+            "The resulting model is then used in a community pipeline that supports image2image and inpainting, "
+            "so the user can keep elements of their room and change specific parts of the image."
+            ""
+        )
     else:
         make_prompt_row()
 

explanation.py

@@ -27,4 +27,4 @@ def make_regeneration_explanation():
         st.image("content/regen_example.png", caption="Room where all concepts except for 'bed', 'lamp', 'table' are regenerated")
 
 def make_segmentation_explanation():
-    pass
+    pass