Spaces:
Runtime error
Runtime error
Trying to fix error:
Browse files
app.py
CHANGED
@@ -9,10 +9,10 @@ os.environ["CUDA_VISIBLE_DEVICES"] = "0"
|
|
9 |
|
10 |
def summarize(text):
|
11 |
# Setting to use the bart-large-cnn model for summarization
|
12 |
-
|
13 |
|
14 |
# To use the t5-base model for summarization:
|
15 |
-
summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base", framework="tf")
|
16 |
|
17 |
summary_text = summarizer(text, max_length=100, min_length=5, do_sample=False)[0]['summary_text']
|
18 |
print(f'Length of initial text: {len(text)}')
|
|
|
9 |
|
10 |
def summarize(text):
|
11 |
# Setting to use the bart-large-cnn model for summarization
|
12 |
+
summarizer = pipeline("summarization")
|
13 |
|
14 |
# To use the t5-base model for summarization:
|
15 |
+
# summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base", framework="tf")
|
16 |
|
17 |
summary_text = summarizer(text, max_length=100, min_length=5, do_sample=False)[0]['summary_text']
|
18 |
print(f'Length of initial text: {len(text)}')
|