Spaces:
Build error
Build error
Commit
·
fc38f97
1
Parent(s):
6290dee
app.py
Browse files
app.py
CHANGED
|
@@ -53,7 +53,7 @@ def predict(image_file, segmentation_png, bitmap_img):
|
|
| 53 |
fake_image = model.predict([latent_vector, final_img_list[2]])
|
| 54 |
fake_img = tf.squeeze(fake_image, axis=0)
|
| 55 |
|
| 56 |
-
return np.array(fake_img)
|
| 57 |
|
| 58 |
# input
|
| 59 |
input = [gr.inputs.Image(type="filepath", label="Ground Truth - Real Image (jpg)"),
|
|
|
|
| 53 |
fake_image = model.predict([latent_vector, final_img_list[2]])
|
| 54 |
fake_img = tf.squeeze(fake_image, axis=0)
|
| 55 |
|
| 56 |
+
return np.array((fake_img+1)/2)
|
| 57 |
|
| 58 |
# input
|
| 59 |
input = [gr.inputs.Image(type="filepath", label="Ground Truth - Real Image (jpg)"),
|