roaringguts
/

DistilBERT

Text Classification

product-categorization

indian-ecommerce

text-embeddings-inference

Model card Files Files and versions

roaringguts commited on Mar 4

Commit

1baebaa

·

verified ·

1 Parent(s): b8861dc

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -24,7 +24,7 @@ DistilBERT fine-tuned to classify Indian fashion product titles into 11 categori
 ```python
 from transformers import pipeline
-clf = pipeline("text-classification", model="roaringguts/your-model-name")
 clf("Nike Dry Fit Running Tshirt")
 # [{'label': 'Activewear', 'score': 0.99}]

 ```python
 from transformers import pipeline
+clf = pipeline("text-classification", model="roaringguts/DistilBERT")
 clf("Nike Dry Fit Running Tshirt")
 # [{'label': 'Activewear', 'score': 0.99}]