xiang-wuu commited on
Commit
9360f41
1 Parent(s): d3ca638

sample image file download has relocated to main app module

Browse files
Files changed (2) hide show
  1. app.py +3 -0
  2. detect_utils.py +0 -4
app.py CHANGED
@@ -1,7 +1,10 @@
1
  import gradio as gr
2
  from classify_utils import classify_tab
3
  from detect_utils import detect_tab
 
4
 
 
 
5
 
6
  if __name__ == '__main__':
7
  gr.TabbedInterface(
 
1
  import gradio as gr
2
  from classify_utils import classify_tab
3
  from detect_utils import detect_tab
4
+ import os
5
 
6
+ os.system("wget https://media.roboflow.com/notebooks/examples/dog.jpeg")
7
+ os.system("wget https://media.roboflow.com/notebooks/examples/dog-2.jpeg")
8
 
9
  if __name__ == '__main__':
10
  gr.TabbedInterface(
detect_utils.py CHANGED
@@ -1,11 +1,7 @@
1
  import gradio as gr
2
  from ultralytics import YOLO
3
- import os
4
  import supervision as sv
5
 
6
- os.system("wget https://media.roboflow.com/notebooks/examples/dog.jpeg")
7
- os.system("wget https://media.roboflow.com/notebooks/examples/dog-2.jpeg")
8
-
9
 
10
  def detect(image, weights, slider_val):
11
  model = YOLO(weights + '.pt')
 
1
  import gradio as gr
2
  from ultralytics import YOLO
 
3
  import supervision as sv
4
 
 
 
 
5
 
6
  def detect(image, weights, slider_val):
7
  model = YOLO(weights + '.pt')