Backend bugfix

backend.py

@@ -5,6 +5,7 @@ from fastapi import FastAPI, UploadFile, File
 import json
 from PIL import Image
 from io import BytesIO
+import numpy as np
 
 from model import get_model
 
@@ -40,7 +41,7 @@ def get_prediction(image):
 	label = np.argmax(probs)
 
 	return {
-		'pred_probs': pred_probs.tolist(),
+		'pred_probs': probs.tolist(),
 		'label': int(label)
 	}
 
@@ -52,8 +53,8 @@ def foo():
 
 
 @app.post("/get_prediction")
-async def predict(face_image: UploadFile = File(...)):
-	image = load_image(await face_image.read())
+async def predict(face_img: UploadFile = File(...)):
+	image = load_image(await face_img.read())
 
 	image = preprocess(image)
 	result = get_prediction(image)