Push model using huggingface_hub.

README.md

@@ -5,11 +5,11 @@ tags:
 - text-classification
 - generated_from_setfit_trainer
 widget:
-- text: I need some modern design assets
-- text: Reduce the letter spacing
-- text: Group the menu items
-- text: Create a photo of a delicious pizza
-- text: Mirror the icon horizontally
+- text: I need some icon suggestions for this layout
+- text: Tighten the letter spacing
+- text: Group the logo and title together
+- text: Create a photo of a mountain landscape
+- text: Mirror the logo vertically
 metrics:
 - accuracy
 pipeline_tag: text-classification
@@ -28,7 +28,7 @@ model-index:
       split: test
     metrics:
     - type: accuracy
-      value: 0.4163860830527497
+      value: 0.29854096520763185
       name: Accuracy
 ---
 
@@ -60,78 +60,78 @@ 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>'Add a corporate presentation background'</li><li>'Insert a modern icon set for the design'</li></ul>       |
-| 1     | <ul><li>'Show me some background options for my design'</li><li>'I need some modern design assets'</li></ul>        |
-| 2     | <ul><li>'Add a footer with contact information'</li><li>'Insert a descriptive caption for the image'</li></ul>      |
-| 3     | <ul><li>'Align the shapes to the top'</li><li>'Right align the buttons'</li></ul>                                   |
-| 4     | <ul><li>'Make the image slide from left'</li><li>'Make the image bounce in'</li></ul>                               |
-| 5     | <ul><li>'Add a spiral animation to the page'</li><li>'Make the page bounce in from the top'</li></ul>               |
-| 6     | <ul><li>'Take out the cat and put a dog in the image'</li><li>"Add glasses to the person's face"</li></ul>          |
-| 7     | <ul><li>'Delete the test element'</li><li>'Remove the unused icon'</li></ul>                                        |
-| 8     | <ul><li>'How do I change the font size?'</li><li>'How can I duplicate elements?'</li></ul>                          |
-| 9     | <ul><li>'Distribute the buttons around the center image'</li><li>'Arrange the shapes in a circle pattern'</li></ul> |
-| 10    | <ul><li>'Create a variation of this design'</li><li>'Make a copy for editing'</li></ul>                             |
-| 11    | <ul><li>'Duplicate the shape and change its color'</li><li>'Copy the image and make it smaller'</li></ul>           |
-| 12    | <ul><li>'Copy the text to the last page'</li><li>'Duplicate the footer to page 2'</li></ul>                         |
-| 13    | <ul><li>'Fix the text layout'</li><li>'Improve the overall typography'</li></ul>                                    |
-| 14    | <ul><li>'Mirror the icon horizontally'</li><li>'Flip the shape vertically'</li></ul>                                |
-| 15    | <ul><li>'Create a photo of a beautiful garden'</li><li>'Create a photo of a delicious pizza'</li></ul>              |
-| 16    | <ul><li>'Generate an Instagram post for a birthday'</li><li>'Generate a Facebook post for a birthday'</li></ul>     |
-| 17    | <ul><li>'Group the icons together'</li><li>'Group the menu items'</li></ul>                                         |
-| 18    | <ul><li>'Position the image at the top right'</li><li>'Move the image to the center'</li></ul>                      |
-| 19    | <ul><li>'Add a dreamy effect'</li><li>'Make the image pop with color'</li></ul>                                     |
-| 20    | <ul><li>'Suggest some shape designs'</li><li>'Suggest some button designs'</li></ul>                                |
-| 21    | <ul><li>'Redo the filter effect'</li><li>'Bring back the original position'</li></ul>                               |
-| 22    | <ul><li>'Remove the background from the icon'</li><li>'Remove the background from the logo'</li></ul>               |
-| 23    | <ul><li>'Delete the watermark'</li><li>'Remove the graffiti'</li></ul>                                              |
-| 24    | <ul><li>'Change the decorative element'</li><li>'Change the background image'</li></ul>                             |
-| 25    | <ul><li>'Update the body text'</li><li>'Change the subtitle'</li></ul>                                              |
-| 26    | <ul><li>'Remove all filters from the image'</li><li>'Restore the original brightness'</li></ul>                     |
-| 27    | <ul><li>'Reduce the shape size'</li><li>'Increase the image size'</li></ul>                                         |
-| 28    | <ul><li>'Change to poster size'</li><li>'Resize to Facebook post dimensions'</li></ul>                              |
-| 29    | <ul><li>'Rotate the element 60 degrees'</li><li>'Rotate the image 90 degrees'</li></ul>                             |
-| 30    | <ul><li>'Randomly place the icons on the page'</li><li>'Randomly place the shapes on the page'</li></ul>            |
-| 31    | <ul><li>'Select the decorative elements'</li><li>'Choose the heading text'</li></ul>                                |
-| 32    | <ul><li>'Make the background transparent'</li><li>'Change to a dark background'</li></ul>                           |
-| 33    | <ul><li>'Change the blend mode to screen'</li><li>'Set the blend mode to darken'</li></ul>                          |
-| 34    | <ul><li>'Add a gaussian blur'</li><li>'Blur the background elements'</li></ul>                                      |
-| 35    | <ul><li>'Add a gradient border'</li><li>'Change the border opacity'</li></ul>                                       |
-| 36    | <ul><li>'Make the image more brilliant'</li><li>'Increase the brightness contrast'</li></ul>                        |
-| 37    | <ul><li>'Send the background to the back'</li><li>'Send the watermark to the back'</li></ul>                        |
-| 38    | <ul><li>'Make the image more dramatic'</li><li>'Make the image more defined'</li></ul>                              |
-| 39    | <ul><li>'Make the image square'</li><li>'Crop the image to a circle'</li></ul>                                      |
-| 40    | <ul><li>'Add a directional shadow'</li><li>'Create a floating shadow'</li></ul>                                     |
-| 41    | <ul><li>'Change the color to silver'</li><li>'Change the button color to purple'</li></ul>                          |
-| 42    | <ul><li>'Increase the body text size'</li><li>'Increase the heading size'</li></ul>                                 |
-| 43    | <ul><li>'Add strikethrough to the text'</li><li>'Add strikethrough to the label'</li></ul>                          |
-| 44    | <ul><li>'Use a serif font for the heading'</li><li>'Change to a clean font'</li></ul>                               |
-| 45    | <ul><li>'Brighten the light areas'</li><li>'Enhance the bright spots'</li></ul>                                     |
-| 46    | <ul><li>'Set the image as full-page background'</li><li>'Make the image the background layer'</li></ul>             |
-| 47    | <ul><li>'Reduce the letter spacing'</li><li>'Add more space between letters'</li></ul>                              |
-| 48    | <ul><li>'Make the text more compact'</li><li>'Reduce the line height'</li></ul>                                     |
-| 49    | <ul><li>'Increase the transparency of the element'</li><li>'Make the element semi-transparent'</li></ul>            |
-| 50    | <ul><li>'Add spacing between text blocks'</li><li>'Make the paragraphs more compact'</li></ul>                      |
-| 51    | <ul><li>'Reduce the color richness'</li><li>'Reduce the color vibrancy'</li></ul>                                   |
-| 52    | <ul><li>'Make the shadows more prominent'</li><li>'Increase the shadow intensity'</li></ul>                         |
-| 53    | <ul><li>'Increase the image resolution'</li><li>'Make the details more sharp'</li></ul>                             |
-| 54    | <ul><li>'Right align the caption'</li><li>'Center the button text'</li></ul>                                        |
-| 55    | <ul><li>'Create a decorative border'</li><li>'Create a border around the text'</li></ul>                            |
-| 56    | <ul><li>'Make the text flow in an arc'</li><li>'Create text that follows a circle'</li></ul>                        |
-| 57    | <ul><li>'Convert the text to a list format'</li><li>'Make the text into a numbered sequence'</li></ul>              |
-| 58    | <ul><li>'Add a dramatic text shadow'</li><li>'Add a colored glow to the text'</li></ul>                             |
-| 59    | <ul><li>'Make the image more amber'</li><li>'Increase the warm color balance'</li></ul>                             |
-| 60    | <ul><li>'Open the file upload dialog'</li><li>'I need to upload a logo'</li></ul>                                   |
-| 61    | <ul><li>'Go back to the original position'</li><li>'Go back to the original text'</li></ul>                         |
-| 62    | <ul><li>'Separate the combined elements'</li><li>'Ungroup the merged elements'</li></ul>                            |
+| Label | Examples                                                           |
+|:------|:-------------------------------------------------------------------|
+| 0     | <ul><li>'Add a corporate presentation background'</li></ul>        |
+| 1     | <ul><li>'I need some icon suggestions for this layout'</li></ul>   |
+| 2     | <ul><li>"Add a heading that says 'Welcome'"</li></ul>              |
+| 3     | <ul><li>'Distribute the shapes across the page'</li></ul>          |
+| 4     | <ul><li>'Add a zoom animation to the logo'</li></ul>               |
+| 5     | <ul><li>'Make everything fade in gradually'</li></ul>              |
+| 6     | <ul><li>'Remove the unwanted text overlay'</li></ul>               |
+| 7     | <ul><li>'Remove the background shape'</li></ul>                    |
+| 8     | <ul><li>'What shape options do you have?'</li></ul>                |
+| 9     | <ul><li>'Distribute the buttons around the center image'</li></ul> |
+| 10    | <ul><li>'Duplicate the page structure'</li></ul>                   |
+| 11    | <ul><li>'Duplicate the icon and move it'</li></ul>                 |
+| 12    | <ul><li>'Copy the content to the final page'</li></ul>             |
+| 13    | <ul><li>'Fix the letter spacing'</li></ul>                         |
+| 14    | <ul><li>'Mirror the logo vertically'</li></ul>                     |
+| 15    | <ul><li>'Create a photo of a mountain landscape'</li></ul>         |
+| 16    | <ul><li>'Create a card for a birthday party'</li></ul>             |
+| 17    | <ul><li>'Group the logo and title together'</li></ul>              |
+| 18    | <ul><li>'Move the image to the center'</li></ul>                   |
+| 19    | <ul><li>'Apply a vintage filter to the photo'</li></ul>            |
+| 20    | <ul><li>'Find me texture patterns'</li></ul>                       |
+| 21    | <ul><li>'Restore the previous color'</li></ul>                     |
+| 22    | <ul><li>'Remove the background from the illustration'</li></ul>    |
+| 23    | <ul><li>'Delete the old sign'</li></ul>                            |
+| 24    | <ul><li>'Replace the old logo with a new one'</li></ul>            |
+| 25    | <ul><li>'Replace the tagline'</li></ul>                            |
+| 26    | <ul><li>'Reset the image adjustments'</li></ul>                    |
+| 27    | <ul><li>'Scale the text up'</li></ul>                              |
+| 28    | <ul><li>'Resize to LinkedIn post dimensions'</li></ul>             |
+| 29    | <ul><li>'Turn the logo 270 degrees'</li></ul>                      |
+| 30    | <ul><li>'Scatter the particles around the text'</li></ul>          |
+| 31    | <ul><li>'Select the footer content'</li></ul>                      |
+| 32    | <ul><li>'Change to a light gray background'</li></ul>              |
+| 33    | <ul><li>'Change the blend mode to color burn'</li></ul>            |
+| 34    | <ul><li>'Blur the logo slightly'</li></ul>                         |
+| 35    | <ul><li>'Add a double border to the image'</li></ul>               |
+| 36    | <ul><li>'Make the photo more illuminated'</li></ul>                |
+| 37    | <ul><li>'Send the logo to the back'</li></ul>                      |
+| 38    | <ul><li>'Increase the image contrast'</li></ul>                    |
+| 39    | <ul><li>'Crop to a polygon'</li></ul>                              |
+| 40    | <ul><li>'Add a gradient shadow'</li></ul>                          |
+| 41    | <ul><li>'Change the background color to green'</li></ul>           |
+| 42    | <ul><li>'Increase the menu font size'</li></ul>                    |
+| 43    | <ul><li>'Make the text bold and strikethrough'</li></ul>           |
+| 44    | <ul><li>'Use a decorative font for the logo'</li></ul>             |
+| 45    | <ul><li>'Brighten the light areas'</li></ul>                       |
+| 46    | <ul><li>'Set the picture as full background'</li></ul>             |
+| 47    | <ul><li>'Tighten the letter spacing'</li></ul>                     |
+| 48    | <ul><li>'Add more line height'</li></ul>                           |
+| 49    | <ul><li>'Reduce the opacity of the overlay'</li></ul>              |
+| 50    | <ul><li>'Reduce the paragraph spacing'</li></ul>                   |
+| 51    | <ul><li>'Make the colors more vibrant'</li></ul>                   |
+| 52    | <ul><li>'Make the shadows deeper'</li></ul>                        |
+| 53    | <ul><li>'Increase the image sharpness'</li></ul>                   |
+| 54    | <ul><li>'Left align the subtitle'</li></ul>                        |
+| 55    | <ul><li>'Create a text container'</li></ul>                        |
+| 56    | <ul><li>'Create text in a spiral pattern'</li></ul>                |
+| 57    | <ul><li>'Convert to bullet point format'</li></ul>                 |
+| 58    | <ul><li>'Add a colored shadow to the text'</li></ul>               |
+| 59    | <ul><li>'Increase the warm color cast'</li></ul>                   |
+| 60    | <ul><li>'I want to upload a new image'</li></ul>                   |
+| 61    | <ul><li>'Revert the opacity change'</li></ul>                      |
+| 62    | <ul><li>'Break apart the grouped elements'</li></ul>               |
 
 ## Evaluation
 
 ### Metrics
 | Label   | Accuracy |
 |:--------|:---------|
-| **all** | 0.4164   |
+| **all** | 0.2985   |
 
 ## Uses
 
@@ -151,7 +151,7 @@ from setfit import SetFitModel
 # Download from the 🤗 Hub
 model = SetFitModel.from_pretrained("setfit_model_id")
 # Run inference
-preds = model("Group the menu items")
+preds = model("Tighten the letter spacing")
 ```
 
 <!--
@@ -183,73 +183,73 @@ preds = model("Group the menu items")
 ### Training Set Metrics
 | Training set | Min | Median | Max |
 |:-------------|:----|:-------|:----|
-| Word count   | 3   | 5.2302 | 11  |
+| Word count   | 3   | 5.2857 | 8   |
 
 | Label | Training Sample Count |
 |:------|:----------------------|
-| 0     | 2                     |
-| 1     | 2                     |
-| 2     | 2                     |
-| 3     | 2                     |
-| 4     | 2                     |
-| 5     | 2                     |
-| 6     | 2                     |
-| 7     | 2                     |
-| 8     | 2                     |
-| 9     | 2                     |
-| 10    | 2                     |
-| 11    | 2                     |
-| 12    | 2                     |
-| 13    | 2                     |
-| 14    | 2                     |
-| 15    | 2                     |
-| 16    | 2                     |
-| 17    | 2                     |
-| 18    | 2                     |
-| 19    | 2                     |
-| 20    | 2                     |
-| 21    | 2                     |
-| 22    | 2                     |
-| 23    | 2                     |
-| 24    | 2                     |
-| 25    | 2                     |
-| 26    | 2                     |
-| 27    | 2                     |
-| 28    | 2                     |
-| 29    | 2                     |
-| 30    | 2                     |
-| 31    | 2                     |
-| 32    | 2                     |
-| 33    | 2                     |
-| 34    | 2                     |
-| 35    | 2                     |
-| 36    | 2                     |
-| 37    | 2                     |
-| 38    | 2                     |
-| 39    | 2                     |
-| 40    | 2                     |
-| 41    | 2                     |
-| 42    | 2                     |
-| 43    | 2                     |
-| 44    | 2                     |
-| 45    | 2                     |
-| 46    | 2                     |
-| 47    | 2                     |
-| 48    | 2                     |
-| 49    | 2                     |
-| 50    | 2                     |
-| 51    | 2                     |
-| 52    | 2                     |
-| 53    | 2                     |
-| 54    | 2                     |
-| 55    | 2                     |
-| 56    | 2                     |
-| 57    | 2                     |
-| 58    | 2                     |
-| 59    | 2                     |
-| 60    | 2                     |
-| 61    | 2                     |
-| 62    | 2                     |
+| 0     | 1                     |
+| 1     | 1                     |
+| 2     | 1                     |
+| 3     | 1                     |
+| 4     | 1                     |
+| 5     | 1                     |
+| 6     | 1                     |
+| 7     | 1                     |
+| 8     | 1                     |
+| 9     | 1                     |
+| 10    | 1                     |
+| 11    | 1                     |
+| 12    | 1                     |
+| 13    | 1                     |
+| 14    | 1                     |
+| 15    | 1                     |
+| 16    | 1                     |
+| 17    | 1                     |
+| 18    | 1                     |
+| 19    | 1                     |
+| 20    | 1                     |
+| 21    | 1                     |
+| 22    | 1                     |
+| 23    | 1                     |
+| 24    | 1                     |
+| 25    | 1                     |
+| 26    | 1                     |
+| 27    | 1                     |
+| 28    | 1                     |
+| 29    | 1                     |
+| 30    | 1                     |
+| 31    | 1                     |
+| 32    | 1                     |
+| 33    | 1                     |
+| 34    | 1                     |
+| 35    | 1                     |
+| 36    | 1                     |
+| 37    | 1                     |
+| 38    | 1                     |
+| 39    | 1                     |
+| 40    | 1                     |
+| 41    | 1                     |
+| 42    | 1                     |
+| 43    | 1                     |
+| 44    | 1                     |
+| 45    | 1                     |
+| 46    | 1                     |
+| 47    | 1                     |
+| 48    | 1                     |
+| 49    | 1                     |
+| 50    | 1                     |
+| 51    | 1                     |
+| 52    | 1                     |
+| 53    | 1                     |
+| 54    | 1                     |
+| 55    | 1                     |
+| 56    | 1                     |
+| 57    | 1                     |
+| 58    | 1                     |
+| 59    | 1                     |
+| 60    | 1                     |
+| 61    | 1                     |
+| 62    | 1                     |
 
 ### Training Hyperparameters
 - batch_size: (64, 64)
@@ -272,11 +272,8 @@ preds = model("Group the menu items")
 ### Training Results
 | Epoch  | Step | Training Loss | Validation Loss |
 |:------:|:----:|:-------------:|:---------------:|
-| 0.0041 | 1    | 0.1097        | -               |
-| 0.2041 | 50   | 0.0477        | -               |
-| 0.4082 | 100  | 0.0106        | -               |
-| 0.6122 | 150  | 0.0045        | -               |
-| 0.8163 | 200  | 0.0032        | -               |
+| 0.0161 | 1    | 0.1282        | -               |
+| 0.8065 | 50   | 0.0118        | -               |
 
 ### Framework Versions
 - Python: 3.12.11

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3c83dbf04e86c45bc32c718b4cffe2fac929e9c674997cd1a149f6cc8e44784a
+oid sha256:63c1f03cab264850c048b0b6eb29d9f8bf69476c716d38f5657dc8b9bd087404
 size 546938168

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:95509daeba7f74aa3e52cc4409d0ae655a5c4092860ba0ee8b19bc3efb7681eb
+oid sha256:8e18a0fa946e9f6dbcb9646d4a936308a4e0e3ddb0886a33bbe2f33e3e8e9215
 size 388927