supritg commited on
Commit
269115a
1 Parent(s): f7bbe78

Added dataset

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,7 +11,7 @@ def yolov8_predict_func(image : gr.Image = None, # uploading an image as input
11
 
12
  # Loading the YOLOv8 model
13
 
14
- model_path = "../Models/YOLOv8 Models/train3-20240514T065222Z-001/train3/weights/best.pt"
15
 
16
  model = YOLO(model_path)
17
 
@@ -51,7 +51,7 @@ outputs = gr.Image(type = "filepath", label = "Output Image")
51
  title = "VPS"
52
 
53
  examples = [
54
- ["../Images/Original_Dataset/ps2.0/testing/indoor-parking lot/007.jpg"]
55
  ]
56
 
57
  yolo_app = gr.Interface(
 
11
 
12
  # Loading the YOLOv8 model
13
 
14
+ model_path = "best.pt"
15
 
16
  model = YOLO(model_path)
17
 
 
51
  title = "VPS"
52
 
53
  examples = [
54
+ ["ps2.0/testing/indoor-parking lot/007.jpg"]
55
  ]
56
 
57
  yolo_app = gr.Interface(