@@ -6,6 +6,7 @@ from decimal import Decimal
 from io import BytesIO
 
 import filetype
+from PIL import Image
 
 from messages.models import Message
 from ml_model.services.EmbeddingService import EmbeddingService
@@ -17,7 +18,6 @@ from payments.selectors.payment_plan_selector import PaymentPlanSelector
 from poller.models import Proxy
 from tools.chats.models import Chat
 from tools.copywrite.models import Copywrite
-from tools.media.models import Image
 from tools.public_api.models import APIStore