Update app.py
Browse files
app.py
CHANGED
|
@@ -4,7 +4,8 @@ from PIL import Image, ImageDraw, ImageFont
|
|
| 4 |
import tempfile
|
| 5 |
|
| 6 |
# Load the YOLOS object detection model
|
| 7 |
-
|
|
|
|
| 8 |
|
| 9 |
# Define some colors to differentiate classes
|
| 10 |
COLORS = ["red", "blue", "green", "orange", "purple", "yellow", "cyan", "magenta"]
|
|
|
|
| 4 |
import tempfile
|
| 5 |
|
| 6 |
# Load the YOLOS object detection model
|
| 7 |
+
|
| 8 |
+
detector = pipeline("object-detection", model="hustvl/yolos-tiny")
|
| 9 |
|
| 10 |
# Define some colors to differentiate classes
|
| 11 |
COLORS = ["red", "blue", "green", "orange", "purple", "yellow", "cyan", "magenta"]
|