Spaces:
Sleeping
Sleeping
muhammadhamza-stack
commited on
Commit
·
bb9a613
1
Parent(s):
10b2c2b
update dockerfile
Browse files- Dockerfile +2 -0
- app.py +1 -1
Dockerfile
CHANGED
|
@@ -10,6 +10,8 @@ WORKDIR /app
|
|
| 10 |
|
| 11 |
# System dependencies (safe defaults)
|
| 12 |
RUN apt-get update && apt-get install -y \
|
|
|
|
|
|
|
| 13 |
git \
|
| 14 |
curl \
|
| 15 |
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
| 10 |
|
| 11 |
# System dependencies (safe defaults)
|
| 12 |
RUN apt-get update && apt-get install -y \
|
| 13 |
+
libgl1 \
|
| 14 |
+
libglib2.0-0 \
|
| 15 |
git \
|
| 16 |
curl \
|
| 17 |
&& rm -rf /var/lib/apt/lists/*
|
app.py
CHANGED
|
@@ -165,7 +165,7 @@ model = YOLO('best1.pt')
|
|
| 165 |
def predict(img, confidence_threshold):
|
| 166 |
# Perform inference
|
| 167 |
# Using verbose=False to suppress unnecessary console output during inference
|
| 168 |
-
results = model(img, verbose=
|
| 169 |
|
| 170 |
# We filter the bounding boxes data array based on the confidence score (index 4)
|
| 171 |
# Filter predictions based on the confidence threshold
|
|
|
|
| 165 |
def predict(img, confidence_threshold):
|
| 166 |
# Perform inference
|
| 167 |
# Using verbose=False to suppress unnecessary console output during inference
|
| 168 |
+
results = model(img, verbose=True)
|
| 169 |
|
| 170 |
# We filter the bounding boxes data array based on the confidence score (index 4)
|
| 171 |
# Filter predictions based on the confidence threshold
|