ckcl commited on
Commit
dd8e6c7
·
verified ·
1 Parent(s): 9e519b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,8 +13,8 @@ class DrowsinessDetector:
13
  self.processor = None
14
  self.input_shape = (224, 224, 3)
15
  self.face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_default.xml')
16
- self.id2label = {0: "不嗜睡", 1: "昏昏欲睡"}
17
- self.label2id = {"不嗜睡": 0, "昏昏欲睡": 1}
18
 
19
  def load_model(self):
20
  """Load the ViT model and processor from Hugging Face Hub"""
 
13
  self.processor = None
14
  self.input_shape = (224, 224, 3)
15
  self.face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_default.xml')
16
+ self.id2label = {0: "Notdrowsy", 1: "drowsy"}
17
+ self.label2id = {"Notdrowsy": 0, "drowsy": 1}
18
 
19
  def load_model(self):
20
  """Load the ViT model and processor from Hugging Face Hub"""