microhum commited on
Commit
fcb5dc8
·
1 Parent(s): 6d8843e

fix image sidebar

Browse files
Files changed (1) hide show
  1. test_few_shot.py +1 -1
test_few_shot.py CHANGED
@@ -81,7 +81,7 @@ def test_main_model(opts):
81
  if opts.streamlit:
82
  st.progress((sample_idx+1)/opts.n_samples, f"Generating Font Sample {sample_idx+1} Please wait...")
83
  im = Image.open(save_file_merge)
84
- set_img(opts.OUTPUT_IMG_KEY, im)
85
  st.image(im, caption=f"sample {sample_idx+1}")
86
 
87
  for char_idx in tqdm(range(opts.char_num)):
 
81
  if opts.streamlit:
82
  st.progress((sample_idx+1)/opts.n_samples, f"Generating Font Sample {sample_idx+1} Please wait...")
83
  im = Image.open(save_file_merge)
84
+ set_img(opts.OUTPUT_IMG_KEY, im.copy())
85
  st.image(im, caption=f"sample {sample_idx+1}")
86
 
87
  for char_idx in tqdm(range(opts.char_num)):