zaidalyafeai · September 7, 2018 08:08
diff --git a/pix2pix_predict.js b/pix2pix_predict.js
 function predict(imgData) {
    return tf.tidy(() => {
        //get the prediction 
        const gImg = model.predict(preprocess(imgData))
        //post process 
        const postImg = postprocess(gImg)
        return postImg   
    })
 }
	function predict(imgData) {
	return tf.tidy(() => {
	//get the prediction
	const gImg = model.predict(preprocess(imgData))
	//post process
	const postImg = postprocess(gImg)
	return postImg
	})
	}