Freak-ppa commited on
Commit
52b4739
·
verified ·
1 Parent(s): 011b14d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -122,6 +122,7 @@ def generate_image(prompt, image):
122
  return Image.open(latest_image)
123
  finally:
124
  delete_image_file(latest_image)
 
125
  time.sleep(1)
126
 
127
  raise TimeoutError("New image was not generated in time")
 
122
  return Image.open(latest_image)
123
  finally:
124
  delete_image_file(latest_image)
125
+ delete_image_file(INPUT_DIR + '/input.png')
126
  time.sleep(1)
127
 
128
  raise TimeoutError("New image was not generated in time")