wangli commited on
Commit
36ce8fe
1 Parent(s): 4868d0a

feat: add reference

Browse files
Files changed (2) hide show
  1. README.md +2 -0
  2. app.py +0 -3
README.md CHANGED
@@ -11,3 +11,5 @@ license: other
11
  ---
12
 
13
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
 
 
11
  ---
12
 
13
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
14
+
15
+ 参考: https://huggingface.co/spaces/ClassCat/YOLOS-Object-Detection
app.py CHANGED
@@ -10,14 +10,11 @@ import matplotlib.patches as patches
10
  import io
11
  from random import choice
12
 
13
-
14
  image_processor_tiny = AutoImageProcessor.from_pretrained("hustvl/yolos-tiny")
15
  model_tiny = AutoModelForObjectDetection.from_pretrained("hustvl/yolos-tiny")
16
 
17
-
18
  import gradio as gr
19
 
20
-
21
  COLORS = ["#ff7f7f", "#ff7fbf", "#ff7fff", "#bf7fff",
22
  "#7f7fff", "#7fbfff", "#7fffff", "#7fffbf",
23
  "#7fff7f", "#bfff7f", "#ffff7f", "#ffbf7f"]
 
10
  import io
11
  from random import choice
12
 
 
13
  image_processor_tiny = AutoImageProcessor.from_pretrained("hustvl/yolos-tiny")
14
  model_tiny = AutoModelForObjectDetection.from_pretrained("hustvl/yolos-tiny")
15
 
 
16
  import gradio as gr
17
 
 
18
  COLORS = ["#ff7f7f", "#ff7fbf", "#ff7fff", "#bf7fff",
19
  "#7f7fff", "#7fbfff", "#7fffff", "#7fffbf",
20
  "#7fff7f", "#bfff7f", "#ffff7f", "#ffbf7f"]