Update app.py

app.py

@@ -3,11 +3,8 @@ import numpy as np
 from ultralytics import YOLO
 import streamlit as st
 
-model = YOLO('MaskDetector.pt')
-font_size = 40
+model = YOLO('LaneSegmentation.pt')
 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')
@@ -15,5 +12,5 @@ if img is not None:
     st.image(img,'Input')
     for result in results:
         for j, mask in result.masks.data:
-            mask = mask.numpy()*255
+            mask = mask.numpy()* 255
             st.image(mask,'Mask')