Spaces:
Runtime error
Runtime error
Alex Strick van Linschoten commited on
Commit ·
02feb96
1
Parent(s): e7c875a
update space name
Browse files
README.md
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
---
|
| 2 |
-
title: Redaction Detector
|
| 3 |
emoji: 📄
|
| 4 |
colorFrom: blue
|
| 5 |
colorTo: yellow
|
|
|
|
| 1 |
---
|
| 2 |
+
title: Redaction Detector for PDFs
|
| 3 |
emoji: 📄
|
| 4 |
colorFrom: blue
|
| 5 |
colorTo: yellow
|
app.py
CHANGED
|
@@ -147,7 +147,7 @@ def predict(pdf, confidence, generate_file):
|
|
| 147 |
pred_dict["img"].save(
|
| 148 |
os.path.join(
|
| 149 |
tmp_dir, filename_without_extension, f"pred-{image}"
|
| 150 |
-
)
|
| 151 |
)
|
| 152 |
# TODO: resize image such that it fits the pdf
|
| 153 |
pdf.image(
|
|
|
|
| 147 |
pred_dict["img"].save(
|
| 148 |
os.path.join(
|
| 149 |
tmp_dir, filename_without_extension, f"pred-{image}"
|
| 150 |
+
),
|
| 151 |
)
|
| 152 |
# TODO: resize image such that it fits the pdf
|
| 153 |
pdf.image(
|