Updated to new model

app.py

@@ -41,7 +41,7 @@ with st.spinner('Loading the models...'):
     )
 
     # Create an inception resnet (in eval mode):
-    resnet = InceptionResnetV1(pretrained='vggface2').eval().to(device)
+    resnet = InceptionResnetV1(pretrained='casia-webface').eval().to(device)
 
     # Define the transformation to preprocess the images
     preprocess = transforms.Compose([
@@ -64,7 +64,7 @@ def extract_features(img):
     return features[0].cpu().numpy()
 
 
-with open("models/svr.p", "rb") as f:
+with open("models/model.p", "rb") as f:
     lr = pickle.load(f)
 
 st.markdown("<center><h1>Know Your BMI</h1></center>", unsafe_allow_html=True)

models/{lr.p → model.p}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1640ed4c6257604fe681b2133f1cbf0ac458888ad8f22ea038db1e1de3866069
-size 4505
+oid sha256:18aa20ae079d7a637a06c3f9b5fefd396d96f1fe343dff6d0372e37ac378c181
+size 12752236

models/svr.p

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:8ccdd3e556fe788c8fed14756678d0da0481396f39c9b36d9c2ece782ec1be0f
-size 12411272