Nous avons eu quelques soucis avec le fonctionnement depuis Hugging Face, donc il est possible de tester notre modèle en local depuis le fichier Light_Novel_Generation.ipynb.
Pour cela il suffit d'aller à la fin du fichier dans la partie Test en local de notre fichier. Et de modifier l'eval_prompt afin que notre modèle le complète.
Running
Pour lancer notre service il suffit de faire
python app.py
Et de se rendre sur l'url indiqué dans le terminal et de générer ce qu'on veut.
Entrainement du modèle
Nous avons entrainé le modèle sur une petite partie du dataset par manque de temps et sur une époque.
Voici la courbe de loss durant le training du modèle
