Update app.py
Browse files
app.py
CHANGED
|
@@ -39,7 +39,10 @@ if use_custom_weights:
|
|
| 39 |
def add_overlay(image: Image.Image, label: str) -> Image.Image:
|
| 40 |
image = image.copy()
|
| 41 |
draw = ImageDraw.Draw(image)
|
| 42 |
-
|
|
|
|
|
|
|
|
|
|
| 43 |
draw.text((10, 10), label, fill="white", font=font)
|
| 44 |
return image
|
| 45 |
|
|
|
|
| 39 |
def add_overlay(image: Image.Image, label: str) -> Image.Image:
|
| 40 |
image = image.copy()
|
| 41 |
draw = ImageDraw.Draw(image)
|
| 42 |
+
try:
|
| 43 |
+
font = ImageFont.load_default()
|
| 44 |
+
except:
|
| 45 |
+
font = None
|
| 46 |
draw.text((10, 10), label, fill="white", font=font)
|
| 47 |
return image
|
| 48 |
|