Skip to content

Commit 2a614c5

Browse files
Rename fire_classifier package, update tags.
1 parent bbc535e commit 2a614c5

File tree

6 files changed

+5
-5
lines changed

6 files changed

+5
-5
lines changed

app/main.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from fastapi import FastAPI, File, UploadFile
2-
from fire_classifier.predictor import ImagePredictor
2+
from classifier.predictor import ImagePredictor
33
from app.settings import API_SETTINGS
44

55
# A: init API
@@ -15,7 +15,7 @@
1515
predictor = ImagePredictor.init_from_config_path(predictor_config_path)
1616

1717

18-
@app.post("/classify-image/", tags=["Detect Fire"])
18+
@app.post("/classify-image/", tags=["Detect Fire in an Image"])
1919
def create_upload_file(file: UploadFile = File(...)):
2020
"""Predicts the possibility that a RBG image contains fire."""
2121
return predictor.predict_from_file(file.file)

app/settings.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
# "tags" parameter of the corresponding app route in main.py!!
1313
API_ENDPOINT_DATA = (
1414
{
15-
"name": "Detect Fire",
15+
"name": "Detect Fire in an Image",
1616
"description": "Predicts the possibility that a color image contains fire.",
1717
},
1818
)
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,8 @@
44
from typing import Dict
55
import yaml
66

7-
from fire_classifier.util import preprocessing
8-
from fire_classifier.util.model import ModelUtility
7+
from classifier.util import preprocessing
8+
from classifier.util.model import ModelUtility
99

1010

1111
class ImagePredictor:

0 commit comments

Comments
 (0)