Final_Assignment_Project

Running

wt002 commited on 6 days ago

Commit

b1f6104

verified ·

1 Parent(s): 358f8ca

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -322,7 +322,7 @@ for task in tasks:
-class BERTEmbeddings(Embedding):
     def __init__(self, model_name='bert-base-uncased'):
         # Load the pre-trained BERT model and tokenizer
         self.tokenizer = BertTokenizer.from_pretrained(model_name)

+class BERTEmbeddings(Embeddings):
     def __init__(self, model_name='bert-base-uncased'):
         # Load the pre-trained BERT model and tokenizer
         self.tokenizer = BertTokenizer.from_pretrained(model_name)