Update Era_s20_updt.py

Era_s20_updt.py

@@ -363,11 +363,11 @@ class Styles_paintings():
 
 def generate_final_image(im1,in_prompt="an oil painting of an baby girl with flowers in a park"):
     paintings = Styles_paintings(in_prompt)
-    paintings.generate_styles()
+    paintings.generate_styles()p
     r_image = im1.resize((512,512))
     print('image shape is',r_image.size)
-    #paintings.generate_styles_with_custom_loss(r_image)
+    paintings.generate_styles_with_custom_loss(r_image)
     
-    #print(len(paintings.output_styles))
+    print(len(paintings.output_styles))
 
-    return [paintings.output_styles[0]], [paintings.output_styles[1]],[paintings.output_styles[2]],[paintings.output_styles[3]],[paintings.output_styles[4]], [paintings.output_styles[0]], [paintings.output_styles[1]],[paintings.output_styles[2]],[paintings.output_styles[3]],[paintings.output_styles[4]]
+    return [paintings.output_styles[0]], [paintings.output_styles[1]],[paintings.output_styles[2]],[paintings.output_styles[3]],[paintings.output_styles[4]], [paintings.generate_styles_with_custom_loss[0]], [paintings.generate_styles_with_custom_loss[1]],[paintings.generate_styles_with_custom_loss[2]],[paintings.generate_styles_with_custom_loss[3]],[paintings.generate_styles_with_custom_loss[4]]