NightPrince commited on
Commit
3737f2c
·
verified ·
1 Parent(s): 2f63d51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,7 +4,7 @@ from ultralytics import YOLO
4
  from PIL import Image
5
 
6
  # Load the trained YOLOv8 model
7
- model = YOLO("/content/runs/detect/train4/weights/best.pt")
8
 
9
  # Define the prediction function
10
  def predict(image):
@@ -17,7 +17,7 @@ interface = gr.Interface(
17
  fn=predict,
18
  inputs=gr.Image(type="pil"),
19
  outputs=gr.Image(type="pil"),
20
- title="Yhaya Detect",
21
  description="Detect Yahya now!"
22
  )
23
 
 
4
  from PIL import Image
5
 
6
  # Load the trained YOLOv8 model
7
+ model = YOLO("yahya-detection.pt")
8
 
9
  # Define the prediction function
10
  def predict(image):
 
17
  fn=predict,
18
  inputs=gr.Image(type="pil"),
19
  outputs=gr.Image(type="pil"),
20
+ title="Yahya Detect",
21
  description="Detect Yahya now!"
22
  )
23