sadimanna commited on
Commit
df2cfd0
1 Parent(s): 6ac35c4

updated app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -29,16 +29,16 @@ ROOT = HERE.parent
29
 
30
  logger = logging.getLogger(__name__)
31
 
32
- ONNX_MODEL_URL = "https://github.com/spmallick/learnopencv/raw/master/Face-Emotion-Recognition/emotion-ferplus-8.onnx" # noqa: E501
33
  ONNX_MODEL_LOCAL_PATH = ROOT / "./emotion-ferplus-8.onnx"
34
- CAFFE_MODEL_URL = "https://github.com/spmallick/learnopencv/raw/master/Face-Emotion-Recognition/RFB-320/RFB-320.caffemodel" # noqa: E501
35
  CAFFE_MODEL_LOCAL_PATH = ROOT / "./RFB-320/RFB-320.caffemodel"
36
  PROTOTXT_URL = "https://github.com/spmallick/learnopencv/raw/master/Facial-Emotion-Recognition/RFB-320/RFB-320.prototxt" # noqa: E501
37
  PROTOTXT_LOCAL_PATH = ROOT / "./RFB-320/RFB-320.prototxt.txt"
38
 
39
- download_file(ONNX_MODEL_URL, ONNX_MODEL_LOCAL_PATH) #, expected_size=None)
40
- download_file(CAFFE_MODEL_URL, CAFFE_MODEL_LOCAL_PATH) #, expected_size=None)
41
- download_file(PROTOTXT_URL, PROTOTXT_LOCAL_PATH) #, expected_size=None)
42
 
43
  # Session-specific caching
44
  onnx_cache_key = "onnx_model"
 
29
 
30
  logger = logging.getLogger(__name__)
31
 
32
+ ONNX_MODEL_URL = "https://github.com/spmallick/learnopencv/raw/master/Facial-Emotion-Recognition/emotion-ferplus-8.onnx" # noqa: E501
33
  ONNX_MODEL_LOCAL_PATH = ROOT / "./emotion-ferplus-8.onnx"
34
+ CAFFE_MODEL_URL = "https://github.com/spmallick/learnopencv/raw/master/Facial-Emotion-Recognition/RFB-320/RFB-320.caffemodel" # noqa: E501
35
  CAFFE_MODEL_LOCAL_PATH = ROOT / "./RFB-320/RFB-320.caffemodel"
36
  PROTOTXT_URL = "https://github.com/spmallick/learnopencv/raw/master/Facial-Emotion-Recognition/RFB-320/RFB-320.prototxt" # noqa: E501
37
  PROTOTXT_LOCAL_PATH = ROOT / "./RFB-320/RFB-320.prototxt.txt"
38
 
39
+ download_file(ONNX_MODEL_URL, ONNX_MODEL_LOCAL_PATH, expected_size=None)
40
+ download_file(CAFFE_MODEL_URL, CAFFE_MODEL_LOCAL_PATH, expected_size=None)
41
+ download_file(PROTOTXT_URL, PROTOTXT_LOCAL_PATH, expected_size=None)
42
 
43
  # Session-specific caching
44
  onnx_cache_key = "onnx_model"