Spaces:
Sleeping
Sleeping
arithescientist
commited on
Commit
·
a39d635
1
Parent(s):
6bc80f7
Update app.py
Browse files
app.py
CHANGED
@@ -36,7 +36,7 @@ def pdf_to_text(PDF, Min):
|
|
36 |
|
37 |
inputs = tokenizer([text], max_length=1024, return_tensors="pt")
|
38 |
|
39 |
-
|
40 |
# Generate Summary
|
41 |
summary_ids = model.generate(inputs["input_ids"], num_beams=2,min_length=Min, max_length=Min+1000)
|
42 |
output_text = tokenizer.batch_decode(summary_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
|
|
|
36 |
|
37 |
inputs = tokenizer([text], max_length=1024, return_tensors="pt")
|
38 |
|
39 |
+
Min = int(Min)
|
40 |
# Generate Summary
|
41 |
summary_ids = model.generate(inputs["input_ids"], num_beams=2,min_length=Min, max_length=Min+1000)
|
42 |
output_text = tokenizer.batch_decode(summary_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
|