Update Era_s20_updt.py
Browse files- 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 |
-
|
|
|
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]]
|