langident
custom_code
impresso-langident / test_floret.py
Gleb Vinarskis
added emas pipeline
ae8276c
raw
history blame
681 Bytes
import floret
model_floret = floret.load_model("LID-40-3-2000000-1-4.bin")
print(type(model_floret))
input_ids = 'this is a text'
input_ids = "En l'an 1348, au plus fort des ravages de la peste noire à travers l'Europe, le Royaume de France se trouvait à la fois au bord du désespoir et face à une opportunité. À la cour du roi Philippe VI, les murs du Louvre étaient animés par les rapports sombres venus de Paris et des villes environnantes. La peste ne montrait aucun signe de répit, et le chancelier Guillaume de Nogaret, le conseiller le plus fidèle du roi, portait le lourd fardeau de gérer la survie du royaume."
print(model_floret.predict([input_ids], k=1))