Update app.py

app.py

@@ -0,0 +1,19 @@
+from PIL import Image,ImageDraw,ImageFont
+import numpy as np
+from ultralytics import YOLO
+import streamlit as st
+
+model = YOLO('MaskDetector.pt')
+font_size = 40
+img = st.file_uploader('Choose an Image')
+font = ImageFont.truetype("arial.ttf", size=font_size)
+arr = ['With mask','Without mask']
+
+if img is not None:
+    img = Image.open(img).resize((640,640)).convert('RGB')
+    results = model(img)
+    st.image(img,'Input')
+    for result in results:
+        for j, mask in result.masks.data:
+            mask = mask.numpy()*255
+            st.image(mask,'Mask')