A

app.py

@@ -13,9 +13,9 @@ import os
 app = FastAPI()
 
 # Mount a static directory for serving images
-folder = os.path.join(os.getcwd(), "static")
+folder = os.path.join(os.getcwd(), "/usr/src/tmp/static")
 os.makedirs(folder)
-app.mount("/static", StaticFiles(directory="static"), name="static")
+app.mount("/usr/src/tmp/static", StaticFiles(directory="usr/src/tmp/static"), name="static")
 
 # Templates for rendering HTML
 templates = Jinja2Templates(directory="templates")
@@ -54,7 +54,7 @@ async def create_upload_file(file: UploadFile = File(...)):
     image = Image.open(io.BytesIO(contents))
     
     # Save the image to a static directory
-    save_path = f"static/{file.filename}"
+    save_path = f"/usr/src/tmp/static/{file.filename}"
     image.save(save_path)
     
     # Perform YOLO prediction
@@ -65,7 +65,7 @@ async def create_upload_file(file: UploadFile = File(...)):
     image_with_boxes = draw_boxes(image_np, predictions)
     
     # Save the image with bounding boxes
-    image_with_boxes_path = f"static/{file.filename.split('.')[0]}_with_boxes.jpg"
+    image_with_boxes_path = f"/usr/src/tmp/static/{file.filename.split('.')[0]}_with_boxes.jpg"
     cv2.imwrite(image_with_boxes_path, cv2.cvtColor(image_with_boxes, cv2.COLOR_RGB2BGR))
     
     # Render the HTML with the image and bounding boxes