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