From 18292e447cfa60cdc8aa0a690996fef1dff40ecf Mon Sep 17 00:00:00 2001
From: cmdr2 <secondary.cmdr2@gmail.com>
Date: Wed, 9 Nov 2022 16:28:58 +0530
Subject: [PATCH] Make the models dir if required

---
 ui/server.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/ui/server.py b/ui/server.py
index bdfbec6c..eb142ddf 100644
--- a/ui/server.py
+++ b/ui/server.py
@@ -240,6 +240,9 @@ def getModels():
 
     def listModels(models_dirname, model_type, model_extensions):
         models_dir = os.path.join(MODELS_DIR, models_dirname)
+        if not os.path.exists(models_dir):
+            os.makedirs(models_dir)
+
         for file in os.listdir(models_dir):
             for model_extension in model_extensions:
                 if file.endswith(model_extension):