muhammadhamza-stack commited on
Commit
bb9a613
·
1 Parent(s): 10b2c2b

update dockerfile

Browse files
Files changed (2) hide show
  1. Dockerfile +2 -0
  2. 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=False)
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