Hugging Face's logo

MonikaG7
/
Video_model

image-classification
Model card Files
xet
Community
MonikaG7 commited on
Commit
4f7808e
·
verified ·
1 Parent(s): 6907f9a

Update inference_video_model.py

Browse files
Files changed (1) hide show
  1. inference_video_model.py +1 -1
inference_video_model.py CHANGED
@@ -1,5 +1,5 @@
1
  from transformers import AutoModelForImageClassification, AutoFeatureExtractor
2
 
3
- model = DrowsinessModel(model_path="MonikaG7/FatiqueModels")
4
  result = model.predict("path_to_video.mp4")
5
  print("Result:", result)
 
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)