diff --git a/endpoints/image.py b/endpoints/image.py
index e46c558..f75fad4 100644
--- a/endpoints/image.py
+++ b/endpoints/image.py
@@ -1,6 +1,7 @@
 from fastapi import UploadFile, Form
 from fastapi.responses import JSONResponse
 import io
+import asyncio
 from PIL import Image
 from pipeline_setup import pipe, IMAGE_TOKEN
 from utils.image_processing import encode_image_base64