Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,8 +2,8 @@ import gradio as gr
|
|
| 2 |
import os
|
| 3 |
|
| 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
|
|
@@ -185,4 +185,4 @@ def detect(img,model):
|
|
| 185 |
return Image.fromarray(im0[:,:,::-1])
|
| 186 |
|
| 187 |
|
| 188 |
-
gr.Interface(detect,[gr.Image(type="pil"),gr.Dropdown(choices=["yolopv2"])], gr.Image(type="pil"),title="Yolopv2",examples=[["
|
|
|
|
| 2 |
import os
|
| 3 |
|
| 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
|
|
|
|
| 185 |
return Image.fromarray(im0[:,:,::-1])
|
| 186 |
|
| 187 |
|
| 188 |
+
gr.Interface(detect,[gr.Image(type="pil"),gr.Dropdown(choices=["yolopv2","yolov7"])], gr.Image(type="pil"),title="Yolopv2",examples=[["horse.jpeg", "yolopv2"]],description="demo for <a href='https://github.com/CAIC-AD/YOLOPv2' 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()
|