Add some TODOs + localhost to allowed_hosts

README.md

@@ -1 +1,9 @@
-# comixify
+# comixify
+
+#TODO:
+- [ ] Prepare readme
+- [ ] Fix model loading to memory
+- [ ] Replace ffmpeg with some keyframes extractor
+- [ ] Prepare own ComixGAN
+- [ ] Prepare some more intelligent method for comic layout
+- [ ] Prepare some method for voice extraction

settings/settings.py

@@ -25,7 +25,7 @@ SECRET_KEY = 'secret_123'
 # SECURITY WARNING: don't run with debug turned on in production!
 DEBUG = True
 
-ALLOWED_HOSTS = ['35.241.250.34', 'comixify.ii.pw.edu.pl']
+ALLOWED_HOSTS = ['35.241.250.34', 'comixify.ii.pw.edu.pl', 'localhost', '127.0.0.1']
 
 
 # Application definition

style_transfer/style_transfer.py

@@ -21,6 +21,7 @@ class StyleTransfer():
         style = kwargs.get("style", "Hayao")
         resize = kwargs.get("resize", 450)
 
+        # TODO: We should load model to memory right after deployment, not on each request.
         # load pretrained model
         model = Transformer()
         model.load_state_dict(torch.load(os.path.join("CartoonGAN/pretrained_model", style + "_net_G_float.pth")))