Update app.py

app.py

@@ -7,6 +7,11 @@ from PIL import Image
 from io import BytesIO
 from ultralytics import YOLO
 from torch.cuda import is_available
+import gdown
+import os
+
+if not os.path.exists('mask_check.pt'):
+    gdown.download(url = "https://drive.google.com/uc?export=download&id=1-HSL744Kbxa3Fc-YZNQPQHFHtxcq340W" output = 'mask_check.pt')
 
 DEMO_IMAGE = 'demo.jpg'
 DEMO_VIDEO = 'demo.mp4'