arithescientist commited on
Commit
a39d635
·
1 Parent(s): 6bc80f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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]