Update predict_model.py

predict_model.py

@@ -13,7 +13,7 @@ def run(in_image):
     in_image = in_image.astype('float32') / 255.0
     predicted = np.clip(model.predict(in_image.reshape(1,160, 160,3)),0.0,1.0).reshape(160, 160,3)
     
-    cv2.imwrite(f'colored-{im_path}', cv2.resize(predicted * 255., (height, width)))
+    # cv2.imwrite(f'colored-{im_path}', cv2.resize(predicted * 255., (height, width)))
     # plot_inference_images(in_image,predicted)
 
     return cv2.resize(predicted * 255., (height, width))