ikraamkb commited on
Commit
5487c8e
·
verified ·
1 Parent(s): 990a952

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -169,7 +169,7 @@ model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
169
  model.eval()
170
  summarizer = pipeline("summarization", model=model, tokenizer=tokenizer, device=-1, batch_size=4)
171
 
172
- reader = easyocr.Reader(['en'], gpu=torch.cuda.is_available())
173
 
174
  summary_cache = {}
175
 
@@ -284,11 +284,11 @@ def create_pdf(summary: str, filename: str):
284
  pdf = FPDF()
285
  pdf.add_page()
286
  pdf.set_font("Arial", 'B', 16)
287
- pdf.cell(200, 10, txt="Document Summary", ln=1, align='C')
288
  pdf.set_font("Arial", size=12)
289
- pdf.cell(200, 10, txt=f"Original file: {filename}", ln=1)
290
  pdf.cell(200, 10, txt=f"Generated on: {datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}", ln=1)
291
  pdf.ln(10)
 
292
  pdf.multi_cell(0, 10, txt=summary)
293
  temp_pdf = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
294
  pdf.output(temp_pdf.name)
 
169
  model.eval()
170
  summarizer = pipeline("summarization", model=model, tokenizer=tokenizer, device=-1, batch_size=4)
171
 
172
+ reader = easyocr.Reader(['en','fr'], gpu=torch.cuda.is_available())
173
 
174
  summary_cache = {}
175
 
 
284
  pdf = FPDF()
285
  pdf.add_page()
286
  pdf.set_font("Arial", 'B', 16)
287
+ pdf.cell(200, 10, txt=f"Summary of {filename}", ln=1, align='C')
288
  pdf.set_font("Arial", size=12)
 
289
  pdf.cell(200, 10, txt=f"Generated on: {datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}", ln=1)
290
  pdf.ln(10)
291
+ pdf.set_font("Arial", size=10)
292
  pdf.multi_cell(0, 10, txt=summary)
293
  temp_pdf = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
294
  pdf.output(temp_pdf.name)