Hugging Face's logo

Spaces:
wookimchye
/
ITI110-Project_Assignment
Runtime error

App Files Community
wookimchye commited on
Commit
38a0343
·
verified ·
1 Parent(s): a2cc3e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,7 +11,8 @@ from torchvision import transforms
11
 
12
  classes = {0: "Defective", 1: "Good"}
13
 
14
- model_path = "best_int8_openvino_model"
 
15
 
16
  def load_model_local():
17
  detection_model = YOLO(model_path, task='classify') # Load the model
 
11
 
12
  classes = {0: "Defective", 1: "Good"}
13
 
14
+ #model_path = "best_int8_openvino_model"
15
+ model_path = "./best.pt"
16
 
17
  def load_model_local():
18
  detection_model = YOLO(model_path, task='classify') # Load the model