Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -189,7 +189,13 @@ def main():
|
|
| 189 |
)
|
| 190 |
|
| 191 |
def create_label_single_pass(background_image, model_data, final_size=(520, 728)):
|
|
|
|
| 192 |
bg_resized = background_image.resize(final_size, Image.Resampling.LANCZOS)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 193 |
draw = ImageDraw.Draw(bg_resized)
|
| 194 |
|
| 195 |
try:
|
|
|
|
| 189 |
)
|
| 190 |
|
| 191 |
def create_label_single_pass(background_image, model_data, final_size=(520, 728)):
|
| 192 |
+
def create_label_single_pass(background_image, model_data, final_size=(520, 728)):
|
| 193 |
bg_resized = background_image.resize(final_size, Image.Resampling.LANCZOS)
|
| 194 |
+
|
| 195 |
+
# If no task is selected (i.e. using default model_data), return the background without drawing any text.
|
| 196 |
+
if not model_data.get("task"):
|
| 197 |
+
return bg_resized
|
| 198 |
+
|
| 199 |
draw = ImageDraw.Draw(bg_resized)
|
| 200 |
|
| 201 |
try:
|