Update gradio_web_demo.py

ensure output_dir exists
This commit is contained in:
Fate_nihility 2024-08-19 10:38:34 +08:00 committed by GitHub
parent 67d34bd826
commit f50e3b95de
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -147,6 +147,8 @@ def delete_old_files():
now = datetime.now()
cutoff = now - timedelta(minutes=10)
output_dir = "./output"
if not os.path.exists(output_dir):
os.makedirs(output_dir)
for filename in os.listdir(output_dir):
file_path = os.path.join(output_dir, filename)
if os.path.isfile(file_path):