multi-lang-rl-model / push_to_hub.py
SivaMallikarjun's picture
Upload push_to_hub.py
e750eb4 verified
raw
history blame
472 Bytes
from transformers import AutoModelForSequenceClassification, AutoTokenizer
# Local model path
model_path = "./models/fine_tuned_xlm_roberta_quantized"
# Load model and tokenizer
model = AutoModelForSequenceClassification.from_pretrained(model_path)
tokenizer = AutoTokenizer.from_pretrained(model_path)
# Save to Hugging Face Hub
model.push_to_hub("SivaMallikarjun/multi-language-rl-model")
tokenizer.push_to_hub("SivaMallikarjun/multi-language-rl-model")