Update README.md

README.md

@@ -45,8 +45,8 @@ Texts are preprocessed with the following rules: [process_transformers](https://
 ```python
 from transformers import AutoModel,AutoProcessor
 from thai2transformers.preprocess import process_transformers
-model = AutoModel.from_pretrained("openthaigpt/CLIPTextCamembertModelWithProjection-contrastive", trust_remote_code=True)
-processor = AutoProcessor.from_pretrained("openthaigpt/CLIPTextCamembertModelWithProjection-contrastive", trust_remote_code=True)
+model = AutoModel.from_pretrained("openthaigpt/CLIPTextCamembertModelWithProjection", trust_remote_code=True)
+processor = AutoProcessor.from_pretrained("openthaigpt/CLIPTextCamembertModelWithProjection", trust_remote_code=True)
 
 input_text = ["This is dog",
               "how are you today",
@@ -79,8 +79,8 @@ image_processor = AutoProcessor.from_pretrained("openai/clip-vit-base-patch32")
 image_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32").to(device)
 
 # Load text model and processor.
-text_processor = AutoProcessor.from_pretrained("openthaigpt/CLIPTextCamembertModelWithProjection-contrastive", trust_remote_code=True)
-text_model = AutoModel.from_pretrained("openthaigpt/CLIPTextCamembertModelWithProjection-contrastive", trust_remote_code=True).to(device)
+text_processor = AutoProcessor.from_pretrained("openthaigpt/CLIPTextCamembertModelWithProjection", trust_remote_code=True)
+text_model = AutoModel.from_pretrained("openthaigpt/CLIPTextCamembertModelWithProjection", trust_remote_code=True).to(device)
 
 class_labels = ['แมว','หมา', 'นก']
 label2id = {label: i for i, label in enumerate(class_labels)}
@@ -118,8 +118,8 @@ image_processor = AutoProcessor.from_pretrained("openai/clip-vit-base-patch32")
 image_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32").to(device)
 
 # Load text model and processor.
-text_processor = AutoProcessor.from_pretrained("openthaigpt/CLIPTextCamembertModelWithProjection-contrastive", trust_remote_code=True)
-text_model = AutoModel.from_pretrained("openthaigpt/CLIPTextCamembertModelWithProjection-contrastive", trust_remote_code=True).to(device)
+text_processor = AutoProcessor.from_pretrained("openthaigpt/CLIPTextCamembertModelWithProjection", trust_remote_code=True)
+text_model = AutoModel.from_pretrained("openthaigpt/CLIPTextCamembertModelWithProjection", trust_remote_code=True).to(device)
 
 text_input = ['แมวสีส้ม','หมาสีดำ', 'นกสีขาว']
 processed_input_text = [process_transformers(input_text_) for input_text_ in input_text ]