Push model using huggingface_hub.

README.md

@@ -4,40 +4,12 @@ tags:
 - sentence-transformers
 - text-classification
 - generated_from_setfit_trainer
-widget:
-- text: "Analyze this clean_code: perms = models.Permission.objects.filter(\n    \
-    \        codename__in=(\"add_customuser\", \"change_customuser\")\n        )\n\
-    \        user.user_permissions.add(*perms)\n        request = self.factory.get(\"\
-    /rand\")\n        request.user = user"
-- text: 'Analyze this hardcoded_secret: SECRET_KEY = "super-secret-jwt-key-do-not-share"  #
-    Django production secret'
-- text: "columns_str = ', '.join(BOOKMARK_COLUMNS)\n    curs.execute('SELECT ' + columns_str\
-    \ + ' FROM bookmarks ' + where_clause\\"
-- text: 'Analyze this hardcoded_secret: const db = new Pool({ password: ''Pr0duct10n#2024''
-    });'
-- text: "Analyze this hardcoded_secret: name:        \"Blake2b with 'Hello, World!'\"\
-    ,\n\t\t\thasher:      NewBlake2B(),\n\t\t\tinput:       []byte(\"Hello, World!\"\
-    ),\n\t\t\texpectedHex: \"511bc81dde11180838c562c82bb35f3223f46061ebde4a955c27b3f489cf1e03\"\
-    ,\n\t\t},\n\t\t{\n\t\t\tname:        \"Blake2b input at max size\","
+widget: []
 metrics:
 - accuracy
 pipeline_tag: text-classification
 library_name: setfit
 inference: true
-model-index:
-- name: SetFit
-  results:
-  - task:
-      type: text-classification
-      name: Text Classification
-    dataset:
-      name: Unknown
-      type: unknown
-      split: test
-    metrics:
-    - type: accuracy
-      value: 0.9760765550239234
-      name: Accuracy
 ---
 
 # SetFit
@@ -67,21 +39,6 @@ The model has been trained using an efficient few-shot learning technique that i
 - **Paper:** [Efficient Few-Shot Learning Without Prompts](https://arxiv.org/abs/2209.11055)
 - **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
 
-### Model Labels
-| Label            | Examples                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                              |
-|:-----------------|:--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
-| SAFE_CODE        | <ul><li>'Analyze this clean_code: },\n                        "summary": "Read Items",\n                        "operationId": "read_items_items__get",\n                        "security": [{"OAuth2PasswordBearer": []}],'</li><li>'Analyze this clean_code: :license: Apache 2.0, see LICENSE for more details.\n"""\n\nfrom __future__ import annotations\n\nimport warnings'</li><li>'Analyze this clean_code: it \'sets rel to stylesheet\' do\n      get \'/style\'\n      expect(headers[\'Link\']).to include(\'rel="stylesheet"\')\n    end\n\n    it \'returns html tag\' do'</li></ul>                                                                                                                                                                                                                                   |
-| VULNERABLE_LOGIC | <ul><li>'try {\n        await exec(`${ezu} apikey-get --apikey hello`);'</li><li>"Analyze this vulnerable_pattern: $user = $_GET[ 'username' ];\n\n\t// Get password\n\t$pass = $_GET[ 'password' ];\n\t$pass = md5( $pass );\n\n\t// Check the database"</li><li>"Analyze this vulnerable_pattern: if( isset( $_GET[ 'Login' ] ) ) {\n\t// Check Anti-CSRF token\n\tcheckToken( $_REQUEST[ 'user_token' ], $_SESSION[ 'session_token' ], 'index.php' );\n\n\t// Sanitise username input\n\t$user = $_GET[ 'username' ];"</li></ul>                                                                                                                                                                                                                                                                                                   |
-| REAL_SECRET      | <ul><li>'Analyze this hardcoded_secret: OPENAI_API_KEY=sk-proj-Tz9mK2nP7qR4sT6uW8vE5fG'</li><li>'Analyze this hardcoded_secret: $this->data = array (\n\t\t\t1 => new User (1, "tony", 0, \'1c8bfe8f801d79745c4631d09fff36c82aa37fc4cce4fc946683d7b336b63032\'),\n\t\t\t2 => new User (2, "morph", 1, \'e5326ba4359f77c2623244acb04f6ac35c4dfca330ebcccdf9b734e5b1df90a8\'),\n\t\t\t3 => new User (3, "chas", 1, \'a89237fc1f9dd8d424d8b8b98b890dbc4a817bfde59af17c39debcc4a14c21de\'),\n\t\t);\n\t\t$this->requestMethod = $requestMethod;\n\t\t$this->userId = $userId;'</li><li>"openai_key: ['Use sk-abcdefghijklmnopqrstuvwxyz0123456789 for calls', 'sk-abcdefghijklmnopqrstuvwxyz0123456789'],\n  github_token: ['Push with ghp_abcdefghijklmnopqrstuvwxyz0123456789', 'ghp_abcdefghijklmnopqrstuvwxyz0123456789'],"</li></ul> |
-| TEST_MOCK        | <ul><li>'Analyze this test_fixture: expected2 = "y=b\'\\\\xe2\\\\xe1\', x=b\'\\\\xe1\\\\xe2\'"\n        self.assertIn(txt, (expected1, expected2))\n\n    def test_text_kwargs(self):\n        txt = str(utils.log_pprint(kwargs={\'x\': \'ŧêßŧ\', \'y\': \'ŧßêŧ\'}))\n        expected1 = "x=\'ŧêßŧ\', y=\'ŧßêŧ\'"\n        expected2 = "y=\'ŧßêŧ\', x=\'ŧêßŧ\'"'</li><li>'const ACCESS_TOKEN = "AKIAIOSFODNN7EXAMPLE"'</li><li>"Analyze this test_fixture: os.pardir,                                      # /path/to/fboy\n)\n\nMEDIA_ROOT = os.path.join(FACTORY_ROOT, 'tmp_test')\n\nDATABASES = {\n    'default': {"</li></ul>                                                                                                                                                                                                  |
-
-## Evaluation
-
-### Metrics
-| Label   | Accuracy |
-|:--------|:---------|
-| **all** | 0.9761   |
-
 ## Uses
 
 ### Direct Use for Inference
@@ -100,7 +57,7 @@ from setfit import SetFitModel
 # Download from the 🤗 Hub
 model = SetFitModel.from_pretrained("setfit_model_id")
 # Run inference
-preds = model("Analyze this hardcoded_secret: const db = new Pool({ password: 'Pr0duct10n#2024' });")
+preds = model("I loved the spiderman movie!")
 ```
 
 <!--
@@ -129,53 +86,6 @@ preds = model("Analyze this hardcoded_secret: const db = new Pool({ password: 'P
 
 ## Training Details
 
-### Training Set Metrics
-| Training set | Min | Median  | Max |
-|:-------------|:----|:--------|:----|
-| Word count   | 1   | 22.5870 | 224 |
-
-| Label            | Training Sample Count |
-|:-----------------|:----------------------|
-| REAL_SECRET      | 113                   |
-| VULNERABLE_LOGIC | 240                   |
-| TEST_MOCK        | 240                   |
-| SAFE_CODE        | 240                   |
-
-### Training Hyperparameters
-- batch_size: (16, 16)
-- num_epochs: (1, 1)
-- max_steps: -1
-- sampling_strategy: oversampling
-- num_iterations: 5
-- body_learning_rate: (2e-05, 1e-05)
-- head_learning_rate: 0.01
-- loss: CosineSimilarityLoss
-- distance_metric: cosine_distance
-- margin: 0.25
-- end_to_end: False
-- use_amp: False
-- warmup_proportion: 0.1
-- l2_weight: 0.01
-- seed: 42
-- eval_max_steps: -1
-- load_best_model_at_end: True
-
-### Training Results
-| Epoch  | Step | Training Loss | Validation Loss |
-|:------:|:----:|:-------------:|:---------------:|
-| 0.0019 | 1    | 0.0334        | -               |
-| 0.0960 | 50   | 0.0839        | -               |
-| 0.1919 | 100  | 0.0608        | -               |
-| 0.2879 | 150  | 0.0444        | -               |
-| 0.3839 | 200  | 0.0285        | -               |
-| 0.4798 | 250  | 0.0267        | -               |
-| 0.5758 | 300  | 0.0143        | -               |
-| 0.6718 | 350  | 0.0141        | -               |
-| 0.7678 | 400  | 0.0116        | -               |
-| 0.8637 | 450  | 0.0099        | -               |
-| 0.9597 | 500  | 0.0092        | -               |
-| 1.0    | 521  | -             | 0.0186          |
-
 ### Framework Versions
 - Python: 3.12.12
 - SetFit: 1.1.3

config_setfit.json

@@ -1,9 +1,9 @@
 {
-  "normalize_embeddings": false,
   "labels": [
     "REAL_SECRET",
     "VULNERABLE_LOGIC",
     "TEST_MOCK",
     "SAFE_CODE"
-  ]
+  ],
+  "normalize_embeddings": false
 }