Update app.py

app.py

@@ -2,7 +2,7 @@ import tensorflow as tf
 import numpy as np
 from imageio.v2 import imread
 import os, glob, cv2, shutil
-# from super_image import EdsrModel, ImageLoader
+from super_image import EdsrModel, ImageLoader
 from PIL import Image
 import gradio as gr
 
@@ -10,7 +10,7 @@ pb = 'dmt.pb'
 style_dim = 8
 img_size=256
 
-# model_scale = EdsrModel.from_pretrained('eugenesiow/edsr-base', scale=2)
+model_scale = EdsrModel.from_pretrained('eugenesiow/edsr-base', scale=2)
 
 def preprocess(img):
   return (img / 255. - 0.5) * 2
@@ -42,13 +42,13 @@ def inference(A,B):
   Xs_ = sess.run(Xs, feed_dict={X: A_img_, Y: B_img_})
   output = deprocess(Xs_)[0]
   output = np.array(np.array(output)*255,dtype=np.uint8)
-  # output = cv2.cvtColor(output, cv2.COLOR_RGB2BGR)
+  output = cv2.cvtColor(output, cv2.COLOR_RGB2BGR)
 
-  # image = Image.fromarray(output)
-  # inputs = ImageLoader.load_image(image)
-  # preds = model_scale(inputs)
-  # ImageLoader.save_image(preds, 'output/scaled_2x.png')
-  return output
+  image = Image.fromarray(output)
+  inputs = ImageLoader.load_image(image)
+  preds = model_scale(inputs)
+  ImageLoader.save_image(preds, 'output/scaled_2x.png')
+  # return output
 
 
 
@@ -62,10 +62,10 @@ def makeupTransfer(arr1,arr2):
   cv2.imwrite("input/ref.png",output2)
   no_makeup = "input/original.png"
   makeup = "input/ref.png"
-  tr_out = inference(no_makeup, makeup)
-  # img = cv2.imread("output/scaled_2x.png")
-  # return cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
-  return tr_out
+  inference(no_makeup, makeup)
+  img = cv2.imread("output/scaled_2x.png")
+  return cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+  # return tr_out
 
 examples = [['faces/no_makeup/xfsy_0226.png', 'faces/makeup/XMY-136.png'],
             ['faces/no_makeup/XYUH-006.png', 'faces/makeup/XMY-266.png'],