xcurvnubaim commited on
Commit
473bf51
·
1 Parent(s): 3c4b4e4

fix: comment unused line

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. main.py +4 -4
Dockerfile CHANGED
@@ -8,7 +8,7 @@ WORKDIR /code
8
  COPY ./requirements.txt /code/requirements.txt
9
 
10
  RUN curl -o models.h5 "https://drive.usercontent.google.com/download?id=1T5HGnk9Mxlb5G6FTxp26BWSrTpzbjtP2&export=download&authuser=0"
11
-
12
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
13
 
14
  COPY . .
 
8
  COPY ./requirements.txt /code/requirements.txt
9
 
10
  RUN curl -o models.h5 "https://drive.usercontent.google.com/download?id=1T5HGnk9Mxlb5G6FTxp26BWSrTpzbjtP2&export=download&authuser=0"
11
+ RUN curl -o best.pt "https://drive.usercontent.google.com/download?id=1qG-tEw5j7P1hPQxTlDrXun32gRp3v3DA&export=download&authuser=0"
12
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
13
 
14
  COPY . .
main.py CHANGED
@@ -109,11 +109,11 @@ def plot_detected_rectangles(image, detections, output_path):
109
  cv2.imwrite(output_path, img_with_rectangles)
110
 
111
 
112
- # Call the animal_detect_and_classify function to get detections
113
- detections = animal_detect_and_classify('/content/cat_tiger.jpg')
114
 
115
- # Plot the detected rectangles with their corresponding class names
116
- plot_detected_rectangles(cv2.imread('/content/cat_tiger.jpg'), detections)
117
 
118
 
119
  @app.post("/predict/v2")
 
109
  cv2.imwrite(output_path, img_with_rectangles)
110
 
111
 
112
+ # # Call the animal_detect_and_classify function to get detections
113
+ # detections = animal_detect_and_classify('/content/cat_tiger.jpg')
114
 
115
+ # # Plot the detected rectangles with their corresponding class names
116
+ # plot_detected_rectangles(cv2.imread('/content/cat_tiger.jpg'), detections)
117
 
118
 
119
  @app.post("/predict/v2")