raicrits
/

BERT_ChangeOfTopic

Text Classification

Model card Files Files and versions

stefanoscotta commited on Jun 4, 2024

Commit

19629f7

·

verified ·

1 Parent(s): 4b45407

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -16,7 +16,7 @@ library_name: transformers
 ---
-# Model Card raicrits/Llama3_ChangeOfTopic
 <!-- Provide a quick summary of what the model is/does. -->
@@ -57,7 +57,7 @@ import torch
 from transformers import AutoTokenizer, BertForSequenceClassification, BertTokenizer, AutoModelForCausalLM, pipeline
-model_bert = torch.load('/opt/data/AI4MEDIA/LLMProject/models/bert_multi_CT_30sec_shift10_weight_loss')
 model_bert = model_bert.to(device_bert)
 tokenizer_bert = AutoTokenizer.from_pretrained('bert-base-multilingual-cased')

 ---
+# Model Card raicrits/BERT_ChangeOfTopic
 <!-- Provide a quick summary of what the model is/does. -->
 from transformers import AutoTokenizer, BertForSequenceClassification, BertTokenizer, AutoModelForCausalLM, pipeline
+model_bert = torch.load('raicrits/BERT_ChangeOfTopic')
 model_bert = model_bert.to(device_bert)
 tokenizer_bert = AutoTokenizer.from_pretrained('bert-base-multilingual-cased')