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(
|