Ayurvedic-Medicine-Recommender / model_download.py
Anupam251272's picture
Create model_download.py
3483007 verified
raw
history blame
453 Bytes
from transformers import pipeline, MBartTokenizer, MBartForConditionalGeneration
from sentence_transformers import SentenceTransformer
# Download models
pipeline("question-answering", model="deepset/xlm-roberta-large-squad2")
SentenceTransformer('paraphrase-multilingual-mpnet-base-v2')
MBartTokenizer.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")