Spaces:
Runtime error
Runtime error
fix missing input
Browse files
app.py
CHANGED
@@ -17,12 +17,17 @@ def get_summary(input_text):
|
|
17 |
summary_model = BartForConditionalGeneration.from_pretrained("ainize/kobart-news")
|
18 |
input_ids = tokenizer.encode(input_text, return_tensors="pt")
|
19 |
summary_text_ids = summary_model.generate(
|
|
|
20 |
input_ids=input_ids,
|
21 |
length_penalty=2,
|
22 |
top_p=0.9,
|
|
|
|
|
|
|
|
|
23 |
max_length=128,
|
24 |
min_length=12,
|
25 |
-
num_beams=
|
26 |
)
|
27 |
# "task_specific_params": {
|
28 |
# "summarization": {
|
|
|
17 |
summary_model = BartForConditionalGeneration.from_pretrained("ainize/kobart-news")
|
18 |
input_ids = tokenizer.encode(input_text, return_tensors="pt")
|
19 |
summary_text_ids = summary_model.generate(
|
20 |
+
<<<<<<< HEAD
|
21 |
input_ids=input_ids,
|
22 |
length_penalty=2,
|
23 |
top_p=0.9,
|
24 |
+
=======
|
25 |
+
length_penalty=1,
|
26 |
+
# top_p=0.9,
|
27 |
+
>>>>>>> d050d191bf6ac2aad382b3d7ee3b1fe6e6302502
|
28 |
max_length=128,
|
29 |
min_length=12,
|
30 |
+
num_beams=2,
|
31 |
)
|
32 |
# "task_specific_params": {
|
33 |
# "summarization": {
|