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')
|