sadimanna commited on
Commit
be6363a
·
1 Parent(s): ed8f39e

updated app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -48,7 +48,9 @@ if onnx_cache_key in st.session_state and caffe_cache_key in st.session_state:
48
  model = st.session_state[onnx_cache_key]
49
  net = st.session_state[caffe_cache_key]
50
  else:
 
51
  model = cv2.dnn.readNetFromONNX(str(ONNX_MODEL_LOCAL_PATH))
 
52
  net = cv2.dnn.readNetFromCaffe(str(PROTOTXT_LOCAL_PATH), str(CAFFE_MODEL_LOCAL_PATH))
53
  st.session_state[onnx_cache_key] = model
54
  st.session_state[caffe_cache_key] = net
 
48
  model = st.session_state[onnx_cache_key]
49
  net = st.session_state[caffe_cache_key]
50
  else:
51
+ # emotion detection model
52
  model = cv2.dnn.readNetFromONNX(str(ONNX_MODEL_LOCAL_PATH))
53
+ # face detection model
54
  net = cv2.dnn.readNetFromCaffe(str(PROTOTXT_LOCAL_PATH), str(CAFFE_MODEL_LOCAL_PATH))
55
  st.session_state[onnx_cache_key] = model
56
  st.session_state[caffe_cache_key] = net