Rust Use #1
by
- opened
Hi, I'm trying to implement this model but I cannot download the pre-treined as I see in the documentation
use rust_bert::resources::RemoteResource;
let model_resource = RemoteResource::from_pretrained((
"distilbert-sst2",
"https://huggingface.co/distilbert-base-uncased-finetuned-sst-2-english/resolve/main/rust_model.ot",
));
I tried:
use rust_bert::resources::RemoteResource;
let model_resource = RemoteResource::from_pretrained((
"ehsanaghaei/SecureBERT",
"https://huggingface.co/ehsanaghaei/SecureBERT/resolve/main/rust_model.ot",
));
But it give me entry not found, some advice?
Hi,
I am not familiar with rust and I am not sure if it can be leveraged to use all HF models. I will take a look at it so stay tuned. In the meanwhile, the best way to download the full working model is to use the transformers package.
from transformers import RobertaTokenizer, RobertaModel
tokenizer = RobertaTokenizer.from_pretrained("ehsanaghaei/SecureBERT")
model = RobertaModel.from_pretrained("ehsanaghaei/SecureBERT")