IDA-SERICS
/

PromptClassifier

Text Classification

databricks-dolly

prompt-category

Model card Files Files and versions

mariadg commited on Apr 16

Commit

6b95560

·

verified ·

1 Parent(s): 3d31fe6

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -48,8 +48,8 @@ The model was evaluated on a balanced version of the dataset. Here are the resul
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
-model = AutoModelForSequenceClassification.from_pretrained("mariadg/deberta-v3-category-classifier")
-tokenizer = AutoTokenizer.from_pretrained("mariadg/deberta-v3-category-classifier")
 text = "The mitochondria is known as the powerhouse of the cell."
 inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)

 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
+model = AutoModelForSequenceClassification.from_pretrained("mariadg/deberta-v3-prompt-recognition")
+tokenizer = AutoTokenizer.from_pretrained("mariadg/deberta-v3-prompt-recognition")
 text = "The mitochondria is known as the powerhouse of the cell."
 inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)