Update Single_image_decod_pt.py