Beasto commited on
Commit
2500880
·
verified ·
1 Parent(s): e0856a3

Update app.py

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