Update app.py

app.py

@@ -8,7 +8,8 @@ import math
 def load_model():
     with open('som.pkl', 'rb') as fid:
         som = pickle.load(fid)
-    return som
+        MM = np.loadtxt('matrizMM.txt', delimiter=" ")
+    return som, MM
 
 def sobel(I):
     m, n = I.shape
@@ -55,24 +56,10 @@ def representativo(imarray):
 
 app = FastAPI()
 
-som = load_model()
-
-MM = np.array([
-       [0., 1., 0., 2., 0., 0., 2., 0., 0., 3.],
-       [3., 1., 2., 1., 2., 3., 0., 1., 3., 3.],
-       [3., 3., 3., 0., 0., 0., 0., 1., 2., 1.],
-       [2., 0., 3., 0., 1., 2., 1., 1., 3., 0.],
-       [1., 1., 2., 2., 2., 1., 1., 1., 2., 2.],
-       [2., 3., 0., 3., 0., 0., 2., 1., 1., 0.],
-       [1., 1., 1., 3., 3., 0., 2., 0., 2., 2.],
-       [1., 2., 2., 3., 3., 1., 2., 1., 3., 1.],
-       [2., 1., 3., 2., 1., 3., 0., 2., 0., 1.],
-       [2., 1., 1., 1., 3., 3., 2., 0., 3., 1.]
-])
-
+som, MM = load_model()
 
 @app.post("/predict/")
-async def predict_fingerprint_api(file: UploadFile = File(...)):
+async def predict(file: UploadFile = File(...)):
     try:
         contents = await file.read()
         image = Image.open(BytesIO(contents)).convert('L')