Zengyf-CVer commited on
Commit
ec8e77a
1 Parent(s): 0931150

app update

Browse files
Files changed (2) hide show
  1. app.py +5 -2
  2. requirements.txt +0 -1
app.py CHANGED
@@ -7,6 +7,7 @@
7
  import argparse
8
  import csv
9
  import json
 
10
  import sys
11
  from pathlib import Path
12
 
@@ -18,6 +19,8 @@ from PIL import Image, ImageDraw, ImageFont
18
  from util.fonts_opt import is_fonts
19
  from util.pdf_opt import pdf_generate
20
 
 
 
21
  ROOT_PATH = sys.path[0] # 根目录
22
 
23
  # 模型路径
@@ -37,7 +40,7 @@ FONTSIZE = 25
37
 
38
 
39
  def parse_args(known=False):
40
- parser = argparse.ArgumentParser(description="Gradio YOLOv5 Det v0.2")
41
  parser.add_argument(
42
  "--model_name", "-mn", default="yolov5s", type=str, help="model name"
43
  )
@@ -320,7 +323,7 @@ def main(args):
320
  outputs = [outputs_img, outputs02_json, outputs03_pdf]
321
 
322
  # 标题
323
- title = "基于Gradio的YOLOv5通用目标检测系统v0.2"
324
  # 描述
325
  description = "<div align='center'>可自定义目标检测模型、安装简单、使用方便</div><div align='center'>Customizable object detection model, easy to install and easy to use</div>"
326
 
 
7
  import argparse
8
  import csv
9
  import json
10
+ import os
11
  import sys
12
  from pathlib import Path
13
 
 
19
  from util.fonts_opt import is_fonts
20
  from util.pdf_opt import pdf_generate
21
 
22
+ os.system("pip install gradio==2.9b40")
23
+
24
  ROOT_PATH = sys.path[0] # 根目录
25
 
26
  # 模型路径
 
40
 
41
 
42
  def parse_args(known=False):
43
+ parser = argparse.ArgumentParser(description="Gradio YOLOv5 Det v0.2.2")
44
  parser.add_argument(
45
  "--model_name", "-mn", default="yolov5s", type=str, help="model name"
46
  )
 
323
  outputs = [outputs_img, outputs02_json, outputs03_pdf]
324
 
325
  # 标题
326
+ title = "基于Gradio的YOLOv5通用目标检测系统v0.2.2"
327
  # 描述
328
  description = "<div align='center'>可自定义目标检测模型、安装简单、使用方便</div><div align='center'>Customizable object detection model, easy to install and easy to use</div>"
329
 
requirements.txt CHANGED
@@ -8,7 +8,6 @@ requests>=2.23.0
8
  scipy>=1.4.1
9
  torch>=1.7.0
10
  torchvision>=0.8.1
11
- gradio==2.9b40
12
  tqdm>=4.41.0
13
  wget>=3.2
14
  rich>=12.2.0
 
8
  scipy>=1.4.1
9
  torch>=1.7.0
10
  torchvision>=0.8.1
 
11
  tqdm>=4.41.0
12
  wget>=3.2
13
  rich>=12.2.0