Spaces:
Runtime error
Runtime error
Commit
·
701e8d1
1
Parent(s):
b90a15d
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,8 +24,8 @@ t5_tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-summariz
|
|
| 24 |
|
| 25 |
def generate_text_summarization(sum_type,article):
|
| 26 |
if article.strip():
|
|
|
|
| 27 |
if sum_type == 'BART Extractive Text Summarization':
|
| 28 |
-
print("text input :",article)
|
| 29 |
inputs = bart_extractive_tokenizer([article], max_length=1024, return_tensors='pt')
|
| 30 |
summary_ids = bart_extractive_model.generate(inputs['input_ids'], num_beams=4, min_length=60, max_length=300, early_stopping=True)
|
| 31 |
|
|
|
|
| 24 |
|
| 25 |
def generate_text_summarization(sum_type,article):
|
| 26 |
if article.strip():
|
| 27 |
+
print("text input :",article)
|
| 28 |
if sum_type == 'BART Extractive Text Summarization':
|
|
|
|
| 29 |
inputs = bart_extractive_tokenizer([article], max_length=1024, return_tensors='pt')
|
| 30 |
summary_ids = bart_extractive_model.generate(inputs['input_ids'], num_beams=4, min_length=60, max_length=300, early_stopping=True)
|
| 31 |
|