Spaces:
Sleeping
Sleeping
BertChristiaens
commited on
Commit
•
5d49ed1
1
Parent(s):
d0ebbc8
front page
Browse files
app.py
CHANGED
@@ -277,41 +277,42 @@ def main():
|
|
277 |
"You can choose between inpainting, segmentation conditioning and re-generating objects, which "
|
278 |
"use our custom trained controlnet model."
|
279 |
)
|
280 |
-
st.
|
281 |
-
|
282 |
-
|
283 |
-
|
284 |
-
|
285 |
-
|
286 |
-
|
287 |
-
|
288 |
-
|
289 |
-
|
290 |
-
|
291 |
-
|
292 |
-
|
293 |
-
|
294 |
-
|
295 |
-
|
296 |
-
|
297 |
-
|
298 |
-
|
299 |
-
|
300 |
-
|
301 |
-
|
302 |
-
|
303 |
-
|
304 |
-
|
305 |
-
|
306 |
-
|
307 |
-
|
308 |
-
|
309 |
-
|
310 |
-
|
311 |
-
|
312 |
-
|
313 |
-
|
314 |
-
|
|
|
315 |
st.write("### Testing images")
|
316 |
st.write("If you don't have any pictures close, you can use one of these images to test the model:")
|
317 |
|
|
|
277 |
"You can choose between inpainting, segmentation conditioning and re-generating objects, which "
|
278 |
"use our custom trained controlnet model."
|
279 |
)
|
280 |
+
with st.expander("Useful information"):
|
281 |
+
st.write("### About the dataset")
|
282 |
+
st.write("To make this demo as good as possible, our team spend a lot of time training a custom model. "
|
283 |
+
"We used the LAION5B dataset to build our custom dataset, which contains 130k images of 15 types of rooms "
|
284 |
+
"in almost 30 design styles. After fetching all these images, we started adding metadata such as "
|
285 |
+
"captions (from the BLIP captioning model) and segmentation maps (from the HuggingFace UperNetForSemanticSegmentation model). "
|
286 |
+
)
|
287 |
+
st.write("### About the model")
|
288 |
+
st.write(
|
289 |
+
"These were then used to train the controlnet model to generate quality interior design images by using "
|
290 |
+
"the segmentation maps and prompts as conditioning information for the model. "
|
291 |
+
"By training on segmentation maps, the enduser has a very finegrained control over which objects they "
|
292 |
+
"want to place in their room. "
|
293 |
+
"The resulting model is then used in a community pipeline that supports image2image and inpainting, "
|
294 |
+
"so the user can keep elements of their room and change specific parts of the image."
|
295 |
+
""
|
296 |
+
)
|
297 |
+
|
298 |
+
st.write("### Trivia")
|
299 |
+
st.write("The first time someone uses the demo after startup, the models still need to be loaded into memory. "
|
300 |
+
"After this initial load, the model is cached as a resource and can be used for all the users. "
|
301 |
+
"To avoid simultaneous requests, we have implemented a queueing mechanism that ensures that only one "
|
302 |
+
"user accesses the model at a time (similar to the Gradio framework).\n"
|
303 |
+
)
|
304 |
+
st.write("To enable the features in the demo, we calculate the underlying segmentation maps and categories that "
|
305 |
+
"are present in the image. This allows us to hide some of the manual work for the user, and "
|
306 |
+
"by doing this, the users don't need to make a segmentation map in an external tool. Everything needed can be done within this demo."
|
307 |
+
)
|
308 |
+
|
309 |
+
st.write("### News: Fondant - an open source data-centric framework for Foundation model finetuning")
|
310 |
+
st.write("The ML6 team is proud to announce that we are open sourcing our Fondant framework, which is a "
|
311 |
+
"data-centric framework that allows you to prepare large scale multimodal datasets with ease. We have implemented the components "
|
312 |
+
"that we used to train this controlnet model in Fondant as an example pipeline, and we are excited to see what you can do with it! In the future we will add a whole library of plug-and-play data preparation components, such as different ML models and filtering steps, in addition to dataset scraping components that connect to LAION5B."
|
313 |
+
)
|
314 |
+
st.write("The framework is build on top of kubeflow pipelines and abstracts all the complexity of efficient storing and moving of large datasets, so you can focus on implemented just that piece of code that you need without worrying about the rest. We also build it to run on each Cloud provider or VM. You can find the code on our github page: https://github.com/ml6team/fondant.")
|
315 |
+
|
316 |
st.write("### Testing images")
|
317 |
st.write("If you don't have any pictures close, you can use one of these images to test the model:")
|
318 |
|