Spaces:
Build error
Build error
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")
|