nkanungo commited on
Commit
9005248
·
1 Parent(s): d023690

Update Era_s20_updt.py

Browse files
Files changed (1) hide show
  1. Era_s20_updt.py +2 -1
Era_s20_updt.py CHANGED
@@ -369,6 +369,7 @@ def generate_final_image(im1,in_prompt):
369
  print('image shape is',r_image.size)
370
  paintings.generate_styles_with_custom_loss(r_image)
371
 
372
- #print(len(paintings.output_styles))
 
373
 
374
  return [paintings.output_styles[0]], [paintings.output_styles[1]],[paintings.output_styles[2]],[paintings.output_styles[3]],[paintings.output_styles[4]],[paintings.output_styles_with_custom_loss[0]],[paintings.output_styles_with_custom_loss[1]],[paintings.output_styles_with_custom_loss[2]],[paintings.output_styles_with_custom_loss[3]],[paintings.output_styles_with_custom_loss[4]]
 
369
  print('image shape is',r_image.size)
370
  paintings.generate_styles_with_custom_loss(r_image)
371
 
372
+ print(len(paintings.output_styles))
373
+ print(len(paintings.output_styles_with_custom_loss))
374
 
375
  return [paintings.output_styles[0]], [paintings.output_styles[1]],[paintings.output_styles[2]],[paintings.output_styles[3]],[paintings.output_styles[4]],[paintings.output_styles_with_custom_loss[0]],[paintings.output_styles_with_custom_loss[1]],[paintings.output_styles_with_custom_loss[2]],[paintings.output_styles_with_custom_loss[3]],[paintings.output_styles_with_custom_loss[4]]