Beasto commited on
Commit
e52841b
·
verified ·
1 Parent(s): 69d3739

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ model = YOLO('DrowsinessDetector.pt')
7
  font_size = 40
8
  img = st.file_uploader('Choose an Image')
9
  font = ImageFont.truetype("arial.ttf", size=font_size)
10
- arr = ['Drowsy','Not Drowsy']
11
 
12
  if img is not None:
13
  img = Image.open(img).resize((640,640)).convert('RGB')
 
7
  font_size = 40
8
  img = st.file_uploader('Choose an Image')
9
  font = ImageFont.truetype("arial.ttf", size=font_size)
10
+ arr = ['Awake','Drowsy']
11
 
12
  if img is not None:
13
  img = Image.open(img).resize((640,640)).convert('RGB')