vichter commited on
Commit
2dd8299
·
verified ·
1 Parent(s): aac34f0

Create download_model.py

Browse files
Files changed (1) hide show
  1. download_model.py +4 -0
download_model.py ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ from transformers import pipeline
2
+ print("Downloading zero-shot classification model...")
3
+ pipeline("zero-shot-classification", model="MoritzLaurer/ModernBERT-large-zeroshot-v2.0")
4
+ print("Model downloaded successfully")