Fix model handling & more

#3

its a little tight fit...

this is basically just copying it from my clone of this hf space-

Scoopala7 changed pull request status to closed

Sign up or log in to comment