Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -4,6 +4,7 @@ import os
|
|
4 |
os.system("wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7.pt")
|
5 |
#os.system("wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-e6e.pt")
|
6 |
os.system("wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-e6.pt")
|
|
|
7 |
|
8 |
import argparse
|
9 |
import time
|
@@ -183,4 +184,4 @@ def detect(img,model):
|
|
183 |
return Image.fromarray(im0[:,:,::-1])
|
184 |
|
185 |
|
186 |
-
gr.Interface(detect,[gr.Image(type="pil"),gr.Dropdown(choices=["yolov7","
|
|
|
4 |
os.system("wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7.pt")
|
5 |
#os.system("wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-e6e.pt")
|
6 |
os.system("wget https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-e6.pt")
|
7 |
+
os.system("wget https://github.com/CAIC-AD/YOLOPv2/releases/download/V0.0.1/yolopv2.pt")
|
8 |
|
9 |
import argparse
|
10 |
import time
|
|
|
184 |
return Image.fromarray(im0[:,:,::-1])
|
185 |
|
186 |
|
187 |
+
gr.Interface(detect,[gr.Image(type="pil"),gr.Dropdown(choices=["yolov7","yolopv2"])], gr.Image(type="pil"),title="Yolov7",examples=[["horses.jpeg", "yolov7"]],description="demo for <a href='https://github.com/WongKinYiu/yolov7' style='text-decoration: underline' target='_blank'>WongKinYiu/yolov7</a> Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors").launch()
|