bgamazay commited on
Commit
d0dc64e
·
verified ·
1 Parent(s): 5bc24b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
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: