Upload 7 files

app.py

@@ -0,0 +1,108 @@
+import streamlit as st
+from PIL import Image
+import torch
+from torchvision import transforms
+from facenet_pytorch import InceptionResnetV1
+import numpy as np
+import json
+import pickle
+from sklearn.metrics.pairwise import cosine_similarity
+from mtcnn import MTCNN
+
+st.set_page_config(page_title="FaceTwin AI", page_icon="🤖", layout="centered")
+
+st.title("FaceTwin AI 🤖")
+st.subheader("Find your closest celebrity match through AI-powered facial similarity 📸 ⭐")
+st.write("---")
+
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
+
+@st.cache_resource(show_spinner="Loading VGGFace2 model...")
+def load_vgg_model():
+    model = InceptionResnetV1(pretrained='vggface2').eval().to(device)
+    return model
+
+@st.cache_resource(show_spinner="Initializing face detector...")
+def load_face_detector():
+    return MTCNN()
+
+@st.cache_data(show_spinner="Loading embeddings and metadata...")
+def load_feature_data():
+    features_list = np.load("features.npy")
+    labels_list = np.load("labels.npy")
+    with open("image_paths.pkl", "rb") as f:
+        imgs_path = pickle.load(f)
+    with open("label_dict.json", "r") as f:
+        label_dict = json.load(f)
+    return features_list, labels_list, imgs_path, label_dict
+
+vgg_feature_extractor = load_vgg_model()
+face_detector = load_face_detector()
+features_list, labels_list, imgs_path, label_dict = load_feature_data()
+
+
+
+transformer = transforms.Compose([
+    transforms.Resize((160, 160)),
+    transforms.ToTensor(),
+    transforms.Normalize(mean=[0.5, 0.5, 0.5],
+                         std=[0.5, 0.5, 0.5])
+])
+
+
+
+
+def predict_actor(img):
+    img = np.array(img)
+    results = face_detector.detect_faces(img)
+
+    if len(results) == 0:
+        return None, None, "No face detected. Please upload a clearer image."
+
+    x, y, width, height = results[0]['box']
+    face = img[y:y + height, x:x + width]
+    face = Image.fromarray(face).convert("RGB")
+
+    img_tensor = transformer(face).unsqueeze(0).to(device)
+
+    with torch.no_grad():
+        features = vgg_feature_extractor(img_tensor)
+
+    flattened_features = features.flatten().cpu().numpy().astype('float16')
+
+    similarities = cosine_similarity([flattened_features], features_list)
+    best_index = np.argmax(similarities)
+
+    label = str(labels_list[best_index])
+    name = label_dict[label]
+    img_path = imgs_path[best_index]
+
+    return name, img_path, None
+
+
+# ===============================
+# 🎛️ STREAMLIT UI
+# ===============================
+upload_img = st.file_uploader("Upload your image", type=["jpg", "jpeg", "png"])
+
+if st.button("Find my Celebrity Match ⭐"):
+    if upload_img is not None:
+        img = Image.open(upload_img)
+
+        with st.spinner("Analyzing your face..."):
+            name, img2_path, error_msg = predict_actor(img)
+
+        if error_msg:
+            st.error(error_msg)
+        else:
+            img2 = Image.open(img2_path)
+            col1, col2 = st.columns(2)
+            with col1:
+                st.image(img, caption="You", use_container_width=True)
+            with col2:
+                st.image(img2, caption=f"Your Celebrity Twin: {name}", use_container_width=True)
+
+            st.success(f"🎉 You look like **{name}**!")
+    else:
+        st.warning("⚠️ Please upload an image first.")

features.npy

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

image_paths.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:820ce83d5a4bad7ebe7f3611f3a23701380e73f5866913c0e0fe39bce0d562cf
+size 790969

label_dict.json

@@ -0,0 +1 @@
+{"0": "Arjun Rampal", "1": "Amitabh Bachchan", "2": "Farhan Akhtar", "3": "Amy Jackson", "4": "Akshay Kumar", "5": "Anil Kapoor", "6": "Abhay Deol", "7": "Abhishek Bachchan", "8": "Emraan Hashmi", "9": "Bipasha Basu", "10": "Deepika Padukone", "11": "Aishwarya Rai", "12": "Arjun Kapoor", "13": "Ileana DCruz", "14": "Huma Qureshi", "15": "Bobby Deol", "16": "Aftab Shivdasani", "17": "Bhumi Pednekar", "18": "Arshad Warsi", "19": "Govinda", "20": "Disha Patani", "21": "Anushka Sharma", "22": "Ajay Devgn", "23": "Akshaye Khanna", "24": "Asin", "25": "Alia Bhatt", "26": "Esha Gupta", "27": "Anushka Shetty", "28": "Amrita Rao", "29": "Hrithik Roshan", "30": "Ameesha Patel", "31": "Ayushmann Khurrana", "32": "Aamir Khan", "33": "Zareen Khan", "34": "Sidharth Malhotra", "35": "Sunny Deol", "36": "Tamannaah Bhatia", "37": "Saif Ali Khan", "38": "Vivek Oberoi", "39": "Varun Dhawan", "40": "Randeep Hooda", "41": "Uday Chopra", "42": "Shraddha Kapoor", "43": "Sonakshi Sinha", "44": "Rani Mukerji", "45": "Richa Chadda", "46": "Yami Gautam", "47": "Vidya Balan", "48": "Shruti Haasan", "49": "Sonam Kapoor", "50": "Suniel Shetty", "51": "Vaani Kapoor", "52": "Salman Khan", "53": "Shahid Kapoor", "54": "Sara Ali Khan", "55": "Riteish Deshmukh", "56": "Taapsee Pannu", "57": "Ranveer Singh", "58": "Shah Rukh Khan", "59": "Tusshar Kapoor", "60": "Shreyas Talpade", "61": "Tiger Shroff", "62": "Sushant Singh Rajput", "63": "Tabu", "64": "Vicky Kaushal", "65": "Shilpa Shetty", "66": "Sanjay Dutt", "67": "Jacqueline Fernandez", "68": "Madhuri Dixit", "69": "Manoj Bajpayee", "70": "Katrina Kaif", "71": "Kajal Aggarwal", "72": "John Abraham", "73": "Lara Dutta", "74": "Priyanka Chopra", "75": "Kareena Kapoor", "76": "Irrfan Khan", "77": "Kartik Aaryan", "78": "Nargis Fakhri", "79": "Kunal Khemu", "80": "Rajkummar Rao", "81": "Prabhas", "82": "Kangana Ranaut", "83": "Naseeruddin Shah", "84": "Nushrat Bharucha", "85": "Kriti Sanon", "86": "Ranbir Kapoor", "87": "Pooja Hegde", "88": "Nana Patekar", "89": "Mrunal Thakur", "90": "Juhi Chawla", "91": "Paresh Rawal", "92": "Kiara Advani", "93": "Kriti Kharbanda", "94": "Preity Zinta", "95": "Prachi Desai", "96": "Parineeti Chopra", "97": "Kajol", "98": "Karisma Kapoor", "99": "R Madhavan"}

labels.npy

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4a056e0d1f5c6cd9fa6c32cd2218fe84c5c3116b81d0e9261fac03778dab8207
+size 17456

requirements.txt

@@ -0,0 +1,10 @@
+streamlit==1.51.0
+facenet-pytorch==2.6.0 
+torch==2.2.2 
+torchvision==0.17.2 
+numpy==1.26.4 
+pillow==10.2.0 
+mtcnn==1.0.0
+pillow==10.2.0
+scikit-learn==1.7.2
+tensorflow==2.19.0