BertChristiaens commited on
Commit
a27c870
1 Parent(s): d24f993
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -328,7 +328,7 @@ def main():
328
  # st.write("The framework is built 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.")
329
 
330
  st.write("### Testing images")
331
- st.write("If you don't have any pictures close, you can use one of these images to test the model:")
332
 
333
  st.session_state['example_image_0'] = Image.open("content/example_0.png")
334
  st.session_state['example_image_1'] = Image.open("content/example_1.jpg")
@@ -354,10 +354,11 @@ def main():
354
  if st.button("Use example 4"):
355
  move_image('example_image_3', 'initial_image', remove_state=True, rerun=True)
356
 
357
- st.write("## Generated examples")
358
- make_image_row(Image.open("content/output_1.png"), Image.open("content/regen_example.png"))
359
- make_image_row(Image.open("content/keep background 2.png"), Image.open("content/output_0.png"))
360
- make_image_row(Image.open("content/segmentation window.png"), Image.open("content/output_3.png"))
 
361
 
362
  else:
363
  make_prompt_row()
 
328
  # st.write("The framework is built 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.")
329
 
330
  st.write("### Testing images")
331
+ st.write("If you don't have any pictures close, you can use one of these images to test the model by clicking on the 'use example X' buttons")
332
 
333
  st.session_state['example_image_0'] = Image.open("content/example_0.png")
334
  st.session_state['example_image_1'] = Image.open("content/example_1.jpg")
 
354
  if st.button("Use example 4"):
355
  move_image('example_image_3', 'initial_image', remove_state=True, rerun=True)
356
 
357
+ with st.expander("Generated examples", expanded=True):
358
+ st.write("## Generated examples")
359
+ make_image_row(Image.open("content/output_1.png"), Image.open("content/regen_example.png"))
360
+ make_image_row(Image.open("content/keep background 2.png"), Image.open("content/output_0.png"))
361
+ make_image_row(Image.open("content/segmentation window.png"), Image.open("content/output_3.png"))
362
 
363
  else:
364
  make_prompt_row()