Push model using huggingface_hub.

README.md

@@ -5,11 +5,11 @@ tags:
 - text-classification
 - generated_from_setfit_trainer
 widget:
-- text: Trouver un vol de Marseille à Berlin, merci.
-- text: La Corée du Nord a-t-elle des alertes de voyage que je devrais être conscientes
-- text: Combien de côtés dans un hexagone
-- text: Quelles sont les alertes pour le Mexique en ce moment ?
-- text: Des alertes de voyage en France sont-elles en vigueur ?
+- text: Quand mon vol décollera-t-il
+- text: Comment dire «hôtel» en finnois
+- text: Qu'est-ce que je suis autorisé à prendre en cabine avec American Airlines
+- text: Où est la meilleure cave du sud de la France
+- text: où devrais-je passer mon temps libre
 metrics:
 - accuracy
 pipeline_tag: text-classification
@@ -46,17 +46,17 @@ The model has been trained using an efficient few-shot learning technique that i
 - **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
 
 ### Model Labels
-| Label | Examples                                                                                                                                                                                                                           |
-|:------|:-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
-| 0     | <ul><li>"S'il vous plaît, aidez-moi à réserver un endroit où séjourner à Pittsburgh du lundi au vendredi"</li><li>'Réservez-moi une chambre du 11 au 15 novembre à Cali'</li><li>'Je souhaite réserver un hôtel à Rome.'</li></ul> |
-| 1     | <ul><li>'Trouvez-moi des vols aller-retour de LAX vers SFOX'</li><li>'Réservez un vol de Chicago à DC lundi et revenant mercredi'</li><li>'Réserver un vol pour New York, aller-retour.'</li></ul>                                 |
-| 2     | <ul><li>'Y a-t-il des alertes de voyage pour la Syrie'</li><li>'La Corée du Nord a-t-elle des alertes de voyage que je devrais être conscientes'</li><li>'Y a-t-il des alertes de voyage pour le Japon ?'</li></ul>                |
-| 8     | <ul><li>'Combien de côtés dans un hexagone'</li><li>'Quand Nintendo a-t-il été créé'</li><li>"Quelle taille d'essuie-glace cette voiture prend-t-elle"</li></ul>                                                                   |
-| 3     | <ul><li>'Y a-t-il une limite de poids de bagage cabine'</li><li>'Quelles sont les règles concernant les bagages à main pour les vols Virgin Airlines'</li><li>'Quelles sont les règles pour les bagages à main?'</li></ul>         |
-| 4     | <ul><li>'Comment dire «hôtel» en finnois'</li><li>"Si j'étais japonais, comment dirais-je que je suis touriste"</li><li>"Pouvez-vous traduire 'bonjour' en espagnol?"</li></ul>                                                    |
-| 5     | <ul><li>'Savez-vous qui je devrais contacter car mes valises ne sont pas arrivées'</li><li>'Je pense que mes bagages ont été perdus'</li><li>"J'ai perdu mes bagages, que dois-je faire?"</li></ul>                                |
-| 6     | <ul><li>"quelle est la porte d'embarquement pour mon vol AF847"</li><li>'Quand mon vol décollera-t-il'</li><li>'Où puis-je vérifier le statut de mon vol?'</li></ul>                                                               |
-| 7     | <ul><li>'Y a-t-il des activités touristiques amusantes en Australie'</li><li>'Donnez-moi une liste de choses à faire à Orlando'</li><li>'Pouvez-vous me suggérer des destinations de voyage?'</li></ul>                            |
+| Label | Examples                                                                                                                                                                                                                                                 |
+|:------|:---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
+| 4     | <ul><li>'Pouvez-vous me dire comment dire «je ne parle pas beaucoup espagnol», en espagnol'</li><li>"Dites-moi comment dire: «C'est une belle matinée» en italien"</li><li>"Si j'étais japonais, comment dirais-je que je suis touriste"</li></ul>       |
+| 2     | <ul><li>'La Corée du Nord a-t-elle des alertes de voyage que je devrais être conscientes'</li><li>'Y a-t-il des alertes de voyage pour la Syrie'</li><li>'Y a-t-il une alerte de voyage pour la Russie'</li></ul>                                        |
+| 6     | <ul><li>'Y a-t-il des nouvelles sur le vol DL123'</li><li>'Quand allons-nous commencer à monter à bord de mon vol prévu'</li><li>'Quand mon vol décollera-t-il'</li></ul>                                                                                |
+| 5     | <ul><li>"Localiser les bagages perdus de Flight America Airlines à O'Hare"</li><li>"Perte des bagages sur Flight America Airlines à O'Hare"</li><li>'Je pense que mes bagages ont été perdus'</li></ul>                                                  |
+| 7     | <ul><li>'Veuillez suggérer des activités touristiques amusantes à Tokyo'</li><li>'Si je vais à Evans, que dois-je faire'</li><li>'Quels sont les bons restaurants à Chicago'</li></ul>                                                                   |
+| 3     | <ul><li>'Voulez-vous me faire part des restrictions de bagage à main pour American Airlines'</li><li>'Restrictions de cabine pour les émirats arabes unis'</li><li>"Qu'est-ce que je suis autorisé à prendre en cabine avec American Airlines"</li></ul> |
+| 0     | <ul><li>'Réservez-moi une chambre du 11 au 15 novembre à Cali'</li><li>'Trouvez-moi un endroit où séjourner à Cali du 11 au 15 novembre'</li><li>'Quels hôtels ont des critiques décentes à Bangor'</li></ul>                                            |
+| 1     | <ul><li>'Trouvez-moi des vols aller-retour de LAX vers SFOX'</li><li>'Réservez un vol de Chicago à DC lundi et revenant mercredi'</li><li>'Je recherche un vol aller-retour de AAF à DIA du 23 février au 28 février'</li></ul>                          |
+| 8     | <ul><li>'Combien coûte des frais de découvert pour la banque'</li><li>"Quelle taille d'essuie-glace cette voiture prend-t-elle"</li><li>'Quel est le plus grand état des États-Unis'</li></ul>                                                           |
 
 ## Uses
 
@@ -76,7 +76,7 @@ from setfit import SetFitModel
 # Download from the 🤗 Hub
 model = SetFitModel.from_pretrained("leaboussekeyt/setfit_model")
 # Run inference
-preds = model("Combien de côtés dans un hexagone")
+preds = model("Quand mon vol décollera-t-il")
 ```
 
 <!--
@@ -108,19 +108,19 @@ preds = model("Combien de côtés dans un hexagone")
 ### Training Set Metrics
 | Training set | Min | Median | Max |
 |:-------------|:----|:-------|:----|
-| Word count   | 4   | 8.4302 | 16  |
+| Word count   | 4   | 9.4    | 20  |
 
 | Label | Training Sample Count |
 |:------|:----------------------|
-| 0     | 10                    |
-| 1     | 10                    |
-| 2     | 10                    |
-| 3     | 10                    |
-| 4     | 10                    |
-| 5     | 10                    |
-| 6     | 10                    |
-| 7     | 10                    |
-| 8     | 6                     |
+| 0     | 7                     |
+| 1     | 6                     |
+| 2     | 5                     |
+| 3     | 8                     |
+| 4     | 7                     |
+| 5     | 7                     |
+| 6     | 6                     |
+| 7     | 8                     |
+| 8     | 21                    |
 
 ### Training Hyperparameters
 - batch_size: (32, 32)
@@ -144,11 +144,10 @@ preds = model("Combien de côtés dans un hexagone")
 ### Training Results
 | Epoch  | Step | Training Loss | Validation Loss |
 |:------:|:----:|:-------------:|:---------------:|
-| 0.0093 | 1    | 0.1927        | -               |
-| 0.4630 | 50   | 0.0828        | -               |
-| 0.9259 | 100  | 0.0139        | -               |
-| 1.3889 | 150  | 0.0055        | -               |
-| 1.8519 | 200  | 0.0036        | -               |
+| 0.0106 | 1    | 0.2933        | -               |
+| 0.5319 | 50   | 0.1645        | -               |
+| 1.0638 | 100  | 0.029         | -               |
+| 1.5957 | 150  | 0.0106        | -               |
 
 ### Framework Versions
 - Python: 3.11.11

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5b35912aa137f70a19e1c3680fbdfd80af3ff299e2b3ce63acc4b5e3e1b1d1d0
+oid sha256:ec4053652ab4b6c2590d0989190c9c02862de515bcb48fe902349664036e7014
 size 442510176

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:24632a4f9949d412497b75a11da28f4a62a1130fddd5035bf1c633e09b687fdb
+oid sha256:9b0f6244c2b21cc5a9fcc6eeddfe633b3b4f35d85a4ee2139e8f796503533f85
 size 56287