Spaces:
Runtime error
Runtime error
Create model_download.py
Browse files- model_download.py +8 -0
model_download.py
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from transformers import pipeline, MBartTokenizer, MBartForConditionalGeneration
|
| 2 |
+
from sentence_transformers import SentenceTransformer
|
| 3 |
+
|
| 4 |
+
# Download models
|
| 5 |
+
pipeline("question-answering", model="deepset/xlm-roberta-large-squad2")
|
| 6 |
+
SentenceTransformer('paraphrase-multilingual-mpnet-base-v2')
|
| 7 |
+
MBartTokenizer.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
|
| 8 |
+
MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
|