add second class in export to make it compatible with inference api 71f9973 Vivien Chappelier commited on Jan 2