azza1625
/

target-identification

Text Classification

target-identification

contrastive-learning

Model card Files Files and versions

azza1625 commited on Oct 21

Commit

cd6c55f

·

verified ·

1 Parent(s): 009afb7

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -49,7 +49,7 @@ class DualEncoderThesisModel(nn.Module, PyTorchModelHubMixin):
         return emb_a, emb_b
-model_name = "ag-charalampous/target-identification"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = DualEncoderThesisModel.from_pretrained(model_name)
@@ -58,7 +58,7 @@ model.eval()
 device = "cpu"
 embeddings_path = hf_hub_download(
-    repo_id="ag-charalampous/target-identification",
     filename="retrieval_data_random_negatives_10_train_data.pkl"
 )

         return emb_a, emb_b
+model_name = "azza1625/target-identification"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = DualEncoderThesisModel.from_pretrained(model_name)
 device = "cpu"
 embeddings_path = hf_hub_download(
+    repo_id="azza1625/target-identification",
     filename="retrieval_data_random_negatives_10_train_data.pkl"
 )