Update app.py
Browse files
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: "
|
| 17 |
-
self.label2id = {"
|
| 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"""
|