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
@@ -44,10 +44,11 @@ def inference(A,B):
   output = np.array(np.array(output)*255,dtype=np.uint8)
   # 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')
+  # image = Image.fromarray(output)
+  # inputs = ImageLoader.load_image(image)
+  # preds = model_scale(inputs)
+  # ImageLoader.save_image(preds, 'output/scaled_2x.png')
+  return output
 
 
 
@@ -61,10 +62,10 @@ def makeupTransfer(arr1,arr2):
   cv2.imwrite("input/ref.png",output2)
   no_makeup = "input/original.png"
   makeup = "input/ref.png"
-  inference(no_makeup, makeup)
-  img = cv2.imread("output/scaled_2x.png")
-  return cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
-
+  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'],