BAJUKA's picture
Update inference code
1c8864c verified