Nisuga Sandira Jayawardana commited on
Commit
dda687a
·
1 Parent(s): ae04b19

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -2
README.md CHANGED
@@ -22,11 +22,11 @@ It achieves the following results on the evaluation set:
22
 
23
  ## Model description
24
 
25
- This model can categorize a given food product title into "PLANT_BASED" or "ANIMAL_BASED".
26
 
27
  ## Intended uses & limitations
28
 
29
- "whey" is an "ANIMAL_BASED" product derived from Cow's milk. Therefore it must categorized as as an ANIMAL_BASED food product.
30
 
31
  Example usage:
32
  ```
@@ -35,6 +35,8 @@ from transformers import pipeline
35
  text = "whey"
36
  classifier = pipeline("text-classification", model="nish-j/food_type_classification_model")
37
  classifier(text)
 
 
38
  ```
39
  ## Training and evaluation data
40
 
 
22
 
23
  ## Model description
24
 
25
+ This model can categorize a given food product title into Plant-based ("PLANT_BASED") or Animal-based("ANIMAL_BASED").
26
 
27
  ## Intended uses & limitations
28
 
29
+ "whey" is an "ANIMAL_BASED" product derived from Cow's milk. Therefore it must be categorized as as an ANIMAL_BASED food product.
30
 
31
  Example usage:
32
  ```
 
35
  text = "whey"
36
  classifier = pipeline("text-classification", model="nish-j/food_type_classification_model")
37
  classifier(text)
38
+
39
+ >>> [{'label': 'ANIMAL_BASED', 'score': 0.9941352605819702}]
40
  ```
41
  ## Training and evaluation data
42