BeMerciless commited on
Commit
06b8b80
1 Parent(s): f559c6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -16,8 +16,6 @@ from utils.torch_utils import select_device, load_classifier, time_synchronized,
16
 
17
 
18
  os.system('git clone https://github.com/WongKinYiu/yolov7')
19
- os.system('wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7.pt')
20
- #os.system('wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-mask.pt')
21
 
22
  #model='best'
23
  def Custom_detect(img,mode):
@@ -164,8 +162,8 @@ inp = gr.Image(type="pil")
164
  inp2= gr.Dropdown(choices=['Custom-Detection','Yolov7-model-detection'])
165
  output = gr.Image(type="pil")
166
 
167
- examples=[["Examples/Image1.jpg","Image1"],["Examples/Image14.jpg","Image14"],["Examples/Image32.jpg","Image32"]]
168
 
169
- io=gr.Interface(fn=Custom_detect, inputs=[inp,inp2], outputs=output, title='Vehicle Detection With Custom YOLOv7')
170
  io.launch()
171
 
 
16
 
17
 
18
  os.system('git clone https://github.com/WongKinYiu/yolov7')
 
 
19
 
20
  #model='best'
21
  def Custom_detect(img,mode):
 
162
  inp2= gr.Dropdown(choices=['Custom-Detection','Yolov7-model-detection'])
163
  output = gr.Image(type="pil")
164
 
165
+ examples=[["Examples/Image1.jpg","Image1"],["Examples/Image2.jpg","Image2"]]
166
 
167
+ io=gr.Interface(fn=Custom_detect, inputs=[inp,inp2], outputs=output, title='Port Structure Detection With Custom YOLOv7')
168
  io.launch()
169