Spaces:
Runtime error
Runtime error
xcurvnubaim
commited on
Commit
·
473bf51
1
Parent(s):
3c4b4e4
fix: comment unused line
Browse files- Dockerfile +1 -1
- 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")
|