Spaces:
Paused
Paused
updated app.py
Browse files
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
|