Zai commited on
Commit
e02dd63
·
1 Parent(s): fbe00aa

Fix huggingface image generation error

Browse files
Files changed (1) hide show
  1. interface.py +1 -1
interface.py CHANGED
@@ -24,4 +24,4 @@ if st.button("Generate Images"):
24
  cols = st.columns(batch_size)
25
  for i, img in enumerate(images):
26
  with cols[i]:
27
- st.image(Image.fromarray(img), use_column_width=True)
 
24
  cols = st.columns(batch_size)
25
  for i, img in enumerate(images):
26
  with cols[i]:
27
+ st.image(img, use_column_width=True)