freak360 commited on
Commit
fb6617f
1 Parent(s): 41b1a4a

Update test.py

Browse files
Files changed (1) hide show
  1. test.py +2 -2
test.py CHANGED
@@ -13,7 +13,7 @@ os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"
13
 
14
 
15
  # Load your trained model
16
- model = YOLO(r"C:\Users\Falcon\Downloads\Projects\Falcon Projects\Drone-Detection-System\drone_detection_system\best.pt")
17
 
18
  def predict_image(image):
19
  # Perform detection
@@ -77,7 +77,7 @@ with gr.Blocks() as demo:
77
  with gr.Tab("Introduction"):
78
  gr.Markdown("**This Application helps in detection of DRONES in an IMAGE, VIDEO or from your WEBCAM depending on your App mode.**")
79
  gr.Markdown("You Don't Necessarily need a Drone to run this app; you can use an image from google.\n\n**SAMPLE OUTPUT:**")
80
- gr.Video("C:\\Users\\Falcon\\Downloads\\Projects\\Falcon Projects\\Drone-Detection-System\\Drone_Detection_Using_YOLOv5\\Drone Detection.mp4", width=800, height=600)
81
  with gr.Tab("Upload Image"):
82
  image_input = gr.Image()
83
  image_output = gr.Image()
 
13
 
14
 
15
  # Load your trained model
16
+ model = YOLO("best.pt")
17
 
18
  def predict_image(image):
19
  # Perform detection
 
77
  with gr.Tab("Introduction"):
78
  gr.Markdown("**This Application helps in detection of DRONES in an IMAGE, VIDEO or from your WEBCAM depending on your App mode.**")
79
  gr.Markdown("You Don't Necessarily need a Drone to run this app; you can use an image from google.\n\n**SAMPLE OUTPUT:**")
80
+ gr.Video("Drone Detection.mp4", width=800, height=600)
81
  with gr.Tab("Upload Image"):
82
  image_input = gr.Image()
83
  image_output = gr.Image()