Prasanna1622 commited on
Commit
518bedc
·
verified ·
1 Parent(s): 9e4e41a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -5,6 +5,11 @@ import gradio as gr
5
  import uuid
6
  import os
7
  import torch
 
 
 
 
 
8
 
9
 
10
  # Load your YOLOv8 model
 
5
  import uuid
6
  import os
7
  import torch
8
+ from ultralytics import YOLO
9
+ import torch.serialization
10
+
11
+ with torch.serialization.safe_globals([YOLO]):
12
+ model = YOLO("best.pt")
13
 
14
 
15
  # Load your YOLOv8 model