file upload fix

masteusz · Mar 1, 2021 · 865bbbe · 865bbbe
1 parent 34558fc
commit 865bbbe
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -1,4 +1,5 @@
 import logging
+from pathlib import Path
 
 from flask import Flask
 
@@ -18,6 +19,10 @@
 app.config["CONFIDENCE"] = config.get("default_confidence", 0.90)
 app.register_blueprint(detector_blueprint)
 
+# Creating upload folder in case it doesn't exist
+Path(app.config["UPLOAD_FOLDER"]).mkdir(exist_ok=True)
+
+# Preloading model
 get_detector()
 
 if __name__ == "__main__":

diff --git a/static/.empty b/static/.empty