vichter commited on
Commit
d1f4ff5
·
verified ·
1 Parent(s): 0a9e95e

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 summarization model...")
3
+ pipeline("summarization", model="Falconsai/text_summarization")
4
+ print("Model downloaded successfully")