Spaces:
Running
Running
rashid996958
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -42,7 +42,7 @@ def inference(A,B):
|
|
42 |
Xs_ = sess.run(Xs, feed_dict={X: A_img_, Y: B_img_})
|
43 |
output = deprocess(Xs_)[0]
|
44 |
output = np.array(np.array(output)*255,dtype=np.uint8)
|
45 |
-
output = cv2.cvtColor(output, cv2.COLOR_RGB2BGR)
|
46 |
|
47 |
image = Image.fromarray(output)
|
48 |
inputs = ImageLoader.load_image(image)
|
|
|
42 |
Xs_ = sess.run(Xs, feed_dict={X: A_img_, Y: B_img_})
|
43 |
output = deprocess(Xs_)[0]
|
44 |
output = np.array(np.array(output)*255,dtype=np.uint8)
|
45 |
+
# output = cv2.cvtColor(output, cv2.COLOR_RGB2BGR)
|
46 |
|
47 |
image = Image.fromarray(output)
|
48 |
inputs = ImageLoader.load_image(image)
|