Spaces:
Runtime error
Runtime error
Update model.py
Browse files
model.py
CHANGED
@@ -30,8 +30,8 @@ class Abstractive_Summarization_Model:
|
|
30 |
model = BartForConditionalGeneration.from_pretrained('Yale-LILY/brio-cnndm-uncased')
|
31 |
tokenizer = BartTokenizer.from_pretrained('Yale-LILY/brio-cnndm-uncased')
|
32 |
else:
|
33 |
-
model = PegasusForConditionalGeneration.from_pretrained('
|
34 |
-
tokenizer = PegasusTokenizer.from_pretrained('
|
35 |
print('[INFO]: Model Successfully Loaded :)')
|
36 |
return model, tokenizer
|
37 |
|
|
|
30 |
model = BartForConditionalGeneration.from_pretrained('Yale-LILY/brio-cnndm-uncased')
|
31 |
tokenizer = BartTokenizer.from_pretrained('Yale-LILY/brio-cnndm-uncased')
|
32 |
else:
|
33 |
+
model = PegasusForConditionalGeneration.from_pretrained('Yale-LILY/brio-xsum-cased')
|
34 |
+
tokenizer = PegasusTokenizer.from_pretrained('Yale-LILY/brio-xsum-cased')
|
35 |
print('[INFO]: Model Successfully Loaded :)')
|
36 |
return model, tokenizer
|
37 |
|