Spaces:
Sleeping
Sleeping
Update test.py
Browse files
test.py
CHANGED
@@ -13,7 +13,7 @@ os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"
|
|
13 |
|
14 |
|
15 |
# Load your trained model
|
16 |
-
model = YOLO(
|
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("
|
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()
|