Darshan03
/

AI-Hackathon

Text Classification

Model card Files Files and versions

Darshan03 commited on Aug 7, 2024

Commit

ff0d4e3

·

verified ·

1 Parent(s): 7f53e02

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -56,8 +56,8 @@ import torch
 from transformers import BertTokenizer, BertForSequenceClassification
 # Load the model and tokenizer
-model = BertForSequenceClassification.from_pretrained('your-username/bert-fine-tuned')
-tokenizer = BertTokenizer.from_pretrained('your-username/bert-fine-tuned')
 # Tokenize the input text
 inputs = tokenizer("Your text here", return_tensors='pt', padding=True, truncation=True, max_length=128)

 from transformers import BertTokenizer, BertForSequenceClassification
 # Load the model and tokenizer
+model = BertForSequenceClassification.from_pretrained('Darshan03/AI-Hackathon')
+tokenizer = BertTokenizer.from_pretrained('Darshan03/AI-Hackathon')
 # Tokenize the input text
 inputs = tokenizer("Your text here", return_tensors='pt', padding=True, truncation=True, max_length=128)