helenai
/

Alireza1044-albert-base-v2-sst2-ov

Text Classification

Inference Endpoints

Model card Files Files and versions Community

helenai commited on May 22

Commit

87cb4a9

•

1 Parent(s): a55d0bd

commit files to HF hub

Files changed (2) hide show

README.md +1 -1
inference.py +1 -1

README.md CHANGED Viewed

@@ -17,7 +17,7 @@ from transformers import AutoTokenizer, pipeline
 # model_id should be set to either a local directory or a model available on the HuggingFace hub.
 model_id = "helenai/Alireza1044-albert-base-v2-sst2-ov"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = OVModelForSequenceClassification.from_pretrained(model_id, revision=')
 pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
 result = pipe("hello world")
 print(result)

 # model_id should be set to either a local directory or a model available on the HuggingFace hub.
 model_id = "helenai/Alireza1044-albert-base-v2-sst2-ov"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
+model = OVModelForSequenceClassification.from_pretrained(model_id)
 pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
 result = pipe("hello world")
 print(result)

inference.py CHANGED Viewed

@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, pipeline
 # model_id should be set to either a local directory or a model available on the HuggingFace hub.
 model_id = "helenai/Alireza1044-albert-base-v2-sst2-ov"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = OVModelForSequenceClassification.from_pretrained(model_id, revision=')
 pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
 result = pipe("hello world")
 print(result)

 # model_id should be set to either a local directory or a model available on the HuggingFace hub.
 model_id = "helenai/Alireza1044-albert-base-v2-sst2-ov"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
+model = OVModelForSequenceClassification.from_pretrained(model_id)
 pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
 result = pipe("hello world")
 print(result)