Spaces:
Runtime error
Runtime error
BeMerciless
commited on
Commit
•
06b8b80
1
Parent(s):
f559c6f
Update app.py
Browse files
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/
|
168 |
|
169 |
-
io=gr.Interface(fn=Custom_detect, inputs=[inp,inp2], outputs=output, title='
|
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 |
|