Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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")
|