Update main.py

main.py

@@ -18,6 +18,7 @@ from sklearn.metrics.pairwise import cosine_similarity
 
 BASE_DIR = "saved_data"
 app = FastAPI()
+mobilenet = MobileNetV2(weights="imagenet", include_top=False, pooling='avg')
 
 def orb_sim(img1, img2):
     # ORB