Update app.py

app.py

@@ -150,5 +150,16 @@ with gr.Blocks(css=css) as demo:
         outputs=[result, seed],
     )
 
+import os
+
+# Danh sách thư mục cần tạo
+folders = ["model", "data", "outputs", "logs"]
+
+for folder in folders:
+    os.makedirs(folder, exist_ok=True)  # exist_ok=True để tránh lỗi nếu thư mục đã tồn tại
+
+print("Thư mục đã được tạo thành công!")
+
+
 if __name__ == "__main__":
     demo.launch()