Rename API, update Dockerfile and dependencies.

Author: UPstartDeveloper

Dockerfile

@@ -8,4 +8,4 @@ RUN apt-get install -y libglib2.0-0
 RUN pip install git+https://github.com/UPstartDeveloper/Fire-Detection-API
 EXPOSE 8080
 
-ENTRYPOINT uvicorn main:app --host 0.0.0.0 --port 8080 --workers 1
+ENTRYPOINT uvicorn app.main:app --host 0.0.0.0 --port 8080 --workers 1

app/main.py

@@ -2,13 +2,17 @@
 
 from fire_classifier.predictor import ImagePredictor
 
-app = FastAPI()
+app = FastAPI(
+    title="Fire Detection API", 
+    description="Informs the probability that an image contains fire."
+)
 
 predictor_config_path = "./app/config.yaml"
 
 predictor = ImagePredictor.init_from_config_url(predictor_config_path)
 
 
-@app.post("/scorefile/")
+@app.post("/classify-image/")
 def create_upload_file(file: UploadFile = File(...)):
+    '''Predicts the possibility that a RBG image contains fire.'''
     return predictor.predict_from_file(file.file)

requirements.txt

@@ -1,7 +1,7 @@
 numpy
 tensorflow-cpu
 fastapi==0.49.0
-opencv-python==4.5.3.56
+opencv-python-headless==4.5.3.56
 PyYAML==5.3
 uvicorn
 python-multipart