Fixed outdated functions to keep this prototype working on HuggingFace

d62f6c4

unverified

gfursin commited on 29 days ago