Spaces:
Runtime error
Runtime error
Fix pretrain path
Browse files- modules/model.py +2 -2
modules/model.py
CHANGED
@@ -7,8 +7,8 @@ import sys
|
|
7 |
# longTokenizer = BartTokenizer.from_pretrained('./ml/distilbart-cnn-12-6/', local_files_only=True)
|
8 |
# longModel = BartForConditionalGeneration.from_pretrained('./ml/distilbart-cnn-12-6/', local_files_only=True)
|
9 |
|
10 |
-
shortTokenizer = BartTokenizer.from_pretrained('
|
11 |
-
shortModel = BartForConditionalGeneration.from_pretrained('
|
12 |
|
13 |
longTokenizer = BartTokenizer.from_pretrained('datien228/distilbart-cnn-12-6-ftn-multi_news')
|
14 |
longModel = BartForConditionalGeneration.from_pretrained('datien228/distilbart-cnn-12-6-ftn-multi_news')
|
|
|
7 |
# longTokenizer = BartTokenizer.from_pretrained('./ml/distilbart-cnn-12-6/', local_files_only=True)
|
8 |
# longModel = BartForConditionalGeneration.from_pretrained('./ml/distilbart-cnn-12-6/', local_files_only=True)
|
9 |
|
10 |
+
shortTokenizer = BartTokenizer.from_pretrained('sshleifer/distilbart-xsum-12-6')
|
11 |
+
shortModel = BartForConditionalGeneration.from_pretrained('sshleifer/distilbart-xsum-12-6')
|
12 |
|
13 |
longTokenizer = BartTokenizer.from_pretrained('datien228/distilbart-cnn-12-6-ftn-multi_news')
|
14 |
longModel = BartForConditionalGeneration.from_pretrained('datien228/distilbart-cnn-12-6-ftn-multi_news')
|