MonikaG7 commited on
Commit
20b6a4a
·
verified ·
1 Parent(s): 4f7808e

Update inference_video_model.py

Browse files
Files changed (1) hide show
  1. inference_video_model.py +4 -4
inference_video_model.py CHANGED
@@ -1,5 +1,5 @@
1
- from transformers import AutoModelForImageClassification, AutoFeatureExtractor
 
 
 
2
 
3
- model = DrowsinessModel(model_path="MonikaG7/Video_model")
4
- result = model.predict("path_to_video.mp4")
5
- print("Result:", result)
 
1
+ from drowsiness_model import DrowsinessModel
2
+
3
+ model = DrowsinessModel()
4
+ result = model.predict('your_video.mp4')
5