mertalev il y a 1 an
Parent
commit
660bf6cdc3
2 fichiers modifiés avec 4 ajouts et 3 suppressions
  1. 2 2
      machine-learning/app/main.py
  2. 2 1
      machine-learning/pyproject.toml

+ 2 - 2
machine-learning/app/main.py

@@ -1,13 +1,13 @@
 import asyncio
 import threading
 from concurrent.futures import ThreadPoolExecutor
-import numpy as np
 from typing import Any
 from zipfile import BadZipFile
 
 import faiss
+import numpy as np
 import orjson
-from fastapi import FastAPI, Form, HTTPException, UploadFile, Depends
+from fastapi import Depends, FastAPI, Form, HTTPException, UploadFile
 from fastapi.responses import ORJSONResponse
 from onnxruntime.capi.onnxruntime_pybind11_state import InvalidProtobuf, NoSuchFile  # type: ignore
 from starlette.formparsers import MultiPartParser

+ 2 - 1
machine-learning/pyproject.toml

@@ -91,7 +91,8 @@ module = [
     "torchvision.transforms",
     "aiocache.backends.memory",
     "aiocache.lock",
-    "aiocache.plugins"
+    "aiocache.plugins",
+    "faiss"
 ]
 ignore_missing_imports = true