fix

app.py

@@ -30,7 +30,7 @@ def memory_efficient(model):
         model.enable_vae_slicing()
     except AttributeError:
         print("enable_vae_slicing is not supported.")
-        
+
     # if device == 'cuda':
     #     try:
     #         model.enable_xformers_memory_efficient_attention()
@@ -202,8 +202,8 @@ def style_fn(image_path, style_name, content_text, output_number=1, diffusion_st
                 )[0][1:]
 
                 n_row = 1
-                n_col = len(inf_seeds) + 1 # 원본추가하려면 + 1
-
+                n_col = len(inf_seeds)
+                
                 # make grid
                 grid = create_image_grid(images, n_row, n_col, padding=10)