Update app.py
Browse files
app.py
CHANGED
@@ -34,7 +34,7 @@ def summarize_text(text):
|
|
34 |
inputs = tokenizer(abstract, return_tensors="pt", truncation=True)
|
35 |
|
36 |
# Generate summary
|
37 |
-
summary_ids = model.generate(inputs['input_ids'], num_beams=4, max_length=45, min_length=
|
38 |
summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
|
39 |
|
40 |
return summary
|
|
|
34 |
inputs = tokenizer(abstract, return_tensors="pt", truncation=True)
|
35 |
|
36 |
# Generate summary
|
37 |
+
summary_ids = model.generate(inputs['input_ids'], num_beams=4, max_length=45, min_length=30, early_stopping=True)
|
38 |
summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
|
39 |
|
40 |
return summary
|