Spaces:
Running
Running
Update app.py
Browse files
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="
|
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)
|