Spaces:
Runtime error
Runtime error
Commit
·
9455659
1
Parent(s):
2da8434
Update app.py
Browse files
app.py
CHANGED
|
@@ -13,7 +13,7 @@ import yake
|
|
| 13 |
from summarizer import Summarizer,TransformerSummarizer
|
| 14 |
from transformers import pipelines
|
| 15 |
nltk.download('punkt')
|
| 16 |
-
from transformers import AutoTokenizer, AutoModelForPreTraining, AutoConfig
|
| 17 |
# model_name = 'distilbert-base-uncased'
|
| 18 |
model_name = 'nlpaueb/legal-bert-base-uncased'
|
| 19 |
#model_name = 'laxya007/gpt2_legal'
|
|
|
|
| 13 |
from summarizer import Summarizer,TransformerSummarizer
|
| 14 |
from transformers import pipelines
|
| 15 |
nltk.download('punkt')
|
| 16 |
+
from transformers import AutoTokenizer, AutoModelForPreTraining, AutoConfig, AutoModel
|
| 17 |
# model_name = 'distilbert-base-uncased'
|
| 18 |
model_name = 'nlpaueb/legal-bert-base-uncased'
|
| 19 |
#model_name = 'laxya007/gpt2_legal'
|