Update backend/main.py

backend/main.py

@@ -22,6 +22,7 @@ import torch
 import torchvision.transforms as T
 from ultralytics import YOLO
 import ultralytics.nn.tasks
+import ultralytics.nn.modules.conv
 import mediapipe as mp
 from fastapi import FastAPI, UploadFile, File, HTTPException, BackgroundTasks, Form, Request
 from fastapi.responses import FileResponse, StreamingResponse, JSONResponse
@@ -41,7 +42,8 @@ torch.serialization.add_safe_globals([
     torch.nn.modules.container.ModuleList,
     torch.nn.modules.upsampling.Upsample,
     torch.nn.modules.pooling.MaxPool2d,
-    ultralytics.nn.tasks.DetectionModel
+    ultralytics.nn.tasks.DetectionModel,
+    ultralytics.nn.modules.conv.Conv
 ])