Update video_model.py

video_model.py

@@ -4,8 +4,8 @@ from transformers import AutoModelForImageClassification, AutoFeatureExtractor
 class DrowsinessModel:
     def __init__(self, model_path, threshold=0.15):
         # Инициализация модели и экстрактора признаков
-        self.model = AutoModelForImageClassification.from_pretrained('MonikaG7/VoiceFatique')
-        self.feature_extractor = AutoFeatureExtractor.from_pretrained('MonikaG7/VoiceFatique')
+        self.model = AutoModelForImageClassification.from_pretrained('MonikaG7/fine-tuned_fatique_model')
+        self.feature_extractor = AutoFeatureExtractor.from_pretrained('MonikaG7/fine-tuned_fatique_model')
         self.threshold = threshold
 
     def predict(self, video_path):