flask apache current dir abs

PiJoules · Apr 2, 2016 · f43b028 · f43b028
1 parent 5a266af
commit f43b028
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/__init__.py b/__init__.py
@@ -14,10 +14,12 @@
 import classify
 import base64
 import requests
+import os
 import numpy as np
 
 ALLOWED_EXTENSIONS = set(['png', 'jpg', 'jpeg'])
-COMPARISONS_DIR = "comparisons"
+CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
+COMPARISONS_DIR = os.path.join(CURRENT_DIR, "comparisons")
 
 
 def allowed_file(filename):

diff --git a/classify.py b/classify.py
@@ -11,6 +11,7 @@ def fname_to_name(fname, dirname):
     fname = re.sub('^' + dirname + '\/', '', fname)
     fname = re.sub('\.(jpg|png|jpeg)', '', fname)
     fname = re.sub('-', ' ', fname)
+    fname = re.sub('_', '', fname)
     return fname
 
 

diff --git a/comparisons/pepe_.png b/comparisons/pepe_.png