ganesh435
/

my-first-model

Text Classification

sentiment-analysis

text-embeddings-inference

Model card Files Files and versions

my-first-model / download_model.py

ganesh435's picture

Upload folder using huggingface_hub

abba290 verified 3 months ago

history blame contribute delete

670 Bytes

	from transformers import AutoModelForSequenceClassification, AutoTokenizer

	model_name = "distilbert-base-uncased-finetuned-sst-2-english"

	tokenizer = AutoTokenizer.from_pretrained(model_name)
	model = AutoModelForSequenceClassification.from_pretrained(model_name)

	model.save_pretrained("./")
	tokenizer.save_pretrained("./")
	from transformers import AutoModelForSequenceClassification, AutoTokenizer

	model_name = "distilbert-base-uncased-finetuned-sst-2-english"

	tokenizer = AutoTokenizer.from_pretrained(model_name)
	model = AutoModelForSequenceClassification.from_pretrained(model_name)

	model.save_pretrained("./")
	tokenizer.save_pretrained("./")