Update app.py

app.py

@@ -8,7 +8,6 @@ from IPython.display import Image
 import matplotlib.pyplot as plt
 
 import gradio as gr
-import cv2
 
 sys.path.insert(0, "/StyleGAN2-GANbanales")
 
@@ -55,16 +54,11 @@ def concatenate(img_array):
   zeros.fill(255)
   white_img = zeros
 
-  a = 1
-  for i in img_array:
-    cv2.imwrite('imagenes/' + str(a) + '.png', i)
-    a+=1
-
   # 1 - 2 images
   if len(img_array) <= 2:
     row_img = img_array[0]
     for i in img_array[1:]:
-      row_img = cv2.hconcat([row_img, i])
+      row_img = np.hstack((row_img, i))
     
     final_img = row_img
   
@@ -72,22 +66,16 @@ def concatenate(img_array):
   elif len(img_array) >= 3 and len(img_array) <= 4:
     row1_img = img_array[0]
     for i in img_array[1:2]:
-      row1_img = cv2.hconcat([row1_img, i])
-
-    cv2.imwrite('imagenes/row1.png', row1_img)
+      row1_img = np.hstack((row1_img, i))
     
     row2_img = img_array[2]
     for i in img_array[3:]:
-      row2_img = cv2.hconcat([row2_img, i])
-
-    cv2.imwrite('imagenes/row2_before.png', row2_img)
+      row2_img = np.hstack((row2_img, i))
     
     for i in range(4-len(img_array)):
-      row2_img = cv2.hconcat([row2_img, white_img])
-
-    cv2.imwrite('imagenes/row2_after.png', row2_img)
+      row2_img = np.hstack((row2_img, white_img))
     
-    final_img = cv2.vconcat([row1_img, row2_img])
+    final_img = np.vstack((row1_img, row2_img))
 
   # 5 - 6 images
   elif len(img_array) >= 4 and len(img_array) <= 6: