fixed code
This commit is contained in:
@@ -1,11 +1,10 @@
|
||||
from PIL import Image, UnidentifiedImageError
|
||||
from fastapi import APIRouter, UploadFile, File, Depends, HTTPException
|
||||
|
||||
from PIL import Image, UnidentifiedImageError
|
||||
|
||||
from app.services.image_moderation_service import ImageModerationService
|
||||
from app.core.dependencies import get_image_moderation_service
|
||||
from app.models.response.image_moderation_response import ImageModerationResponse
|
||||
|
||||
from app.models.response.image_moderation_response import \
|
||||
ImageModerationResponse
|
||||
from app.services.image_moderation_service import ImageModerationService
|
||||
|
||||
router = APIRouter(
|
||||
prefix="/api/v1/moderation",
|
||||
@@ -26,7 +25,6 @@ def moderate_image(
|
||||
file: UploadFile = File(...),
|
||||
service: ImageModerationService = Depends(get_image_moderation_service)
|
||||
):
|
||||
|
||||
# 1. Проверка content-type — самая дешёвая, делаем её первой
|
||||
if file.content_type not in ALLOWED_CONTENT_TYPES:
|
||||
raise HTTPException(
|
||||
@@ -63,4 +61,4 @@ def moderate_image(
|
||||
reason=prediction.reason,
|
||||
label=prediction.label,
|
||||
detected_labels=prediction.detected_labels
|
||||
)
|
||||
)
|
||||
|
||||
@@ -6,7 +6,6 @@ from app.models.dto.text_request import TextRequest
|
||||
from app.models.response.moderation_response import ModerationResponse
|
||||
from app.services.text_moderation_service import TextModerationService
|
||||
|
||||
|
||||
from app.core.dependencies import (
|
||||
get_text_moderation_service
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user