shionhonda commited on
Commit
bbce684
·
1 Parent(s): ae3a4af
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -25,9 +25,10 @@ def scale_to_255(x):
25
 
26
  if __name__ == "__main__":
27
  st.title("Sushi Diffusion")
28
- st.markdown("The generation process takes about 10 mins. If you don't want to wait, please visit https://thissushidoesnotexist.com/ for static samples.")
 
29
  model = get_model()
30
- st.markdown('Press the button below to generate sushi!')
31
  if st.button('🍣'):
32
  bar = st.progress(0)
33
  img = torch.randn((1,3,64,64), device="cpu")
 
25
 
26
  if __name__ == "__main__":
27
  st.title("Sushi Diffusion")
28
+ st.text("The generation process takes about 10 mins.")
29
+ st.text("If you don't want to wait, please visit: https://thissushidoesnotexist.com/")
30
  model = get_model()
31
+ st.text('Press the button below to generate sushi!')
32
  if st.button('🍣'):
33
  bar = st.progress(0)
34
  img = torch.randn((1,3,64,64), device="cpu")