multi-lang-rl-model / push_to_hub.py
SivaMallikarjun's picture
Update push_to_hub.py
59a463b verified
raw
history blame
452 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-lang-rl-model")
tokenizer.push_to_hub("SivaMallikarjun/multi-lang-rl-model")