shikibu9419 commited on
Commit
a7d7626
·
1 Parent(s): fbe631d

Fix application config

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. app.py +10 -10
Dockerfile CHANGED
@@ -26,4 +26,4 @@ RUN pip install --no-cache-dir --upgrade -r $HOME/app/requirements.txt
26
 
27
  COPY . $HOME/app
28
 
29
- CMD ["flask", "run", "--host", "0.0.0.0"]
 
26
 
27
  COPY . $HOME/app
28
 
29
+ CMD ["gunicorn", "app:app"]
app.py CHANGED
@@ -54,11 +54,11 @@ def predict(process_name):
54
  img_array = np.asarray(bytearray(imgfile.stream.read()), dtype=np.uint8)
55
  img = cv2.imdecode(img_array, cv2.IMREAD_COLOR)
56
 
57
- if 'colorpatch' in request.files:
58
- patchfile = request.files['colorpatch']
59
- patch_array = np.asarray(bytearray(patchfile.stream.read()), dtype=np.uint8)
60
- colorpatch = cv2.imdecode(colorpatch_array, cv2.IMREAD_COLOR)
61
- update_patch(process_name, colorpatch)
62
 
63
  img = predict_img(process_name, img)
64
 
@@ -77,11 +77,11 @@ def optimize(process_name):
77
  img_array = np.asarray(bytearray(imgfile.stream.read()), dtype=np.uint8)
78
  img = cv2.imdecode(img_array, cv2.IMREAD_COLOR)
79
 
80
- if 'colorpatch' in request.files:
81
- patchfile = request.files['colorpatch']
82
- patch_array = np.asarray(bytearray(patchfile.stream.read()), dtype=np.uint8)
83
- colorpatch = cv2.imdecode(colorpatch_array, cv2.IMREAD_COLOR)
84
- update_patch(process_name, colorpatch)
85
 
86
  (opt_img, preview_img) = optimize_img(process_name, img)
87
 
 
54
  img_array = np.asarray(bytearray(imgfile.stream.read()), dtype=np.uint8)
55
  img = cv2.imdecode(img_array, cv2.IMREAD_COLOR)
56
 
57
+ # if 'colorpatch' in request.files:
58
+ # patchfile = request.files['colorpatch']
59
+ # patch_array = np.asarray(bytearray(patchfile.stream.read()), dtype=np.uint8)
60
+ # colorpatch = cv2.imdecode(colorpatch_array, cv2.IMREAD_COLOR)
61
+ # update_patch(process_name, colorpatch)
62
 
63
  img = predict_img(process_name, img)
64
 
 
77
  img_array = np.asarray(bytearray(imgfile.stream.read()), dtype=np.uint8)
78
  img = cv2.imdecode(img_array, cv2.IMREAD_COLOR)
79
 
80
+ # if 'colorpatch' in request.files:
81
+ # patchfile = request.files['colorpatch']
82
+ # patch_array = np.asarray(bytearray(patchfile.stream.read()), dtype=np.uint8)
83
+ # colorpatch = cv2.imdecode(colorpatch_array, cv2.IMREAD_COLOR)
84
+ # update_patch(process_name, colorpatch)
85
 
86
  (opt_img, preview_img) = optimize_img(process_name, img)
87