From f50e3b95de0f31ac80fb4179a5397a1aea51d067 Mon Sep 17 00:00:00 2001
From: Fate_nihility <61657922+CodeLyokoscj@users.noreply.github.com>
Date: Mon, 19 Aug 2024 10:38:34 +0800
Subject: [PATCH] Update gradio_web_demo.py

ensure output_dir exists
---
 inference/gradio_web_demo.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/inference/gradio_web_demo.py b/inference/gradio_web_demo.py
index b81c0ba..601f9a8 100644
--- a/inference/gradio_web_demo.py
+++ b/inference/gradio_web_demo.py
@@ -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):