Spaces:

DevashishBhake
/

Face_Mask_Detection

Runtime error

DevashishBhake commited on May 8

Commit

9b6527a

•

1 Parent(s): 95ee55a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,9 +1,9 @@
 import gradio as gr
 from PIL import Image
 import numpy as np
-from ultralytics import YOLO
-model = YOLO('./best.pt')
 def yolo(im, size=512):
     g = (size / max(im.size))  # gain

 import gradio as gr
 from PIL import Image
 import numpy as np
+import torch
+model = torch.hub.load('./yolov5', 'custom', path='./best.pt', force_reload=True, source='local')
 def yolo(im, size=512):
     g = (size / max(im.size))  # gain