fixing docker to accept saving files

Dockerfile

@@ -5,7 +5,6 @@ WORKDIR /code
 COPY ./requirements.txt /code/requirements.txt
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
-RUN apt update && apt install -y ffmpeg
 
 RUN useradd -m -u 1000 user
 USER user

server.py

@@ -27,6 +27,7 @@ async def dominant_color(file: UploadFile = File(...), num_colors: int = Form(..
     """
     Receive an image file and an integer and return the dominant color(s).
     """
+    print('num_colors: ', num_colors)
     file_content = await file.read()
     image_bytes = io.BytesIO(file_content)
     im = Image.open(image_bytes)
@@ -97,13 +98,21 @@ async def recolor(file: UploadFile = File(...), colors: str = Form(...),  model:
         recolorPaletteBasedTransfer.recolor(image_np, colors)
     img_file = open("./result.jpg", "rb")
     return StreamingResponse(img_file, media_type="image/jpeg")
+@app.get("/test/")
+async def test():
+    """
+    Test endpoint to check if the server is running.
+    """
+    return {"message": "Server is running"}
+
 
 
 
 
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+    print("Server is running")
+    uvicorn.run(app, host="0.0.0.0", port=7860)