Hugging Face's logo

Spaces:
hysts
/
ibug-emotion_recognition
Running on Zero

App Files Community
hysts HF Staff commited on
Commit
366a48b
·
1 Parent(s): b42c5db

Update

Browse files
Files changed (2) hide show
  1. app.py +3 -1
  2. requirements.txt +3 -3
app.py CHANGED
@@ -27,7 +27,9 @@ face_detector = RetinaFacePredictor(threshold=0.8, device="cpu", model=RetinaFac
27
  face_detector.device = device
28
  face_detector.net.to(device)
29
 
30
- landmark_detector = FANPredictor(device="cpu", model=FANPredictor.get_model("2dfan2"))
 
 
31
  landmark_detector.device = device
32
  landmark_detector.net.to(device)
33
 
 
27
  face_detector.device = device
28
  face_detector.net.to(device)
29
 
30
+ landmark_detector = FANPredictor(
31
+ device="cpu", model=FANPredictor.get_model("2dfan2"), config=FANPredictor.create_config(use_jit=False)
32
+ )
33
  landmark_detector.device = device
34
  landmark_detector.net.to(device)
35
 
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  gradio==4.33.0
2
  numpy==1.26.4
3
- opencv-python-headless==4.10.0.82
4
- torch==2.0.1
5
- torchvision==0.15.2
 
1
  gradio==4.33.0
2
  numpy==1.26.4
3
+ opencv-python==4.10.0.82
4
+ torch==2.2.0
5
+ torchvision==0.17.0