Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ tokenizer = T5Tokenizer.from_pretrained(model_path)
|
|
8 |
|
9 |
|
10 |
|
11 |
-
def summarize_text(text,
|
12 |
# Preprocess the text
|
13 |
inputs = tokenizer.encode(
|
14 |
"summarize: " + text,
|
@@ -21,7 +21,7 @@ def summarize_text(text,max_length_slider):
|
|
21 |
# Generate the summary
|
22 |
summary_ids = model.generate(
|
23 |
inputs,
|
24 |
-
max_length=
|
25 |
num_beams=5,
|
26 |
early_stopping=True,
|
27 |
no_repeat_ngram_size=2,
|
|
|
8 |
|
9 |
|
10 |
|
11 |
+
def summarize_text(text,max_length):
|
12 |
# Preprocess the text
|
13 |
inputs = tokenizer.encode(
|
14 |
"summarize: " + text,
|
|
|
21 |
# Generate the summary
|
22 |
summary_ids = model.generate(
|
23 |
inputs,
|
24 |
+
max_length=max_length,
|
25 |
num_beams=5,
|
26 |
early_stopping=True,
|
27 |
no_repeat_ngram_size=2,
|