Upload 101 files

.gitattributes

@@ -33,3 +33,34 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Aishwarya[[:space:]]Rai.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Amy[[:space:]]Jackson.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Anil[[:space:]]Kapoor.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Asin.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Ayushmann[[:space:]]Khurrana.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Bobby[[:space:]]Deol.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Emraan[[:space:]]Hashmi.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Esha[[:space:]]Gupta.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Juhi[[:space:]]Chawla.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Kajal[[:space:]]Aggarwal.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Karisma[[:space:]]Kapoor.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Kartik[[:space:]]Aaryan.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Katrina[[:space:]]Kaif.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Lara[[:space:]]Dutta.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Madhuri[[:space:]]Dixit.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Mrunal[[:space:]]Thakur.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Naseeruddin[[:space:]]Shah.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Nushrat[[:space:]]Bharucha.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Parineeti[[:space:]]Chopra.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Pooja[[:space:]]Hegde.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Preity[[:space:]]Zinta.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Priyanka[[:space:]]Chopra.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Rani[[:space:]]Mukerji.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Ranveer[[:space:]]Singh.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Saif[[:space:]]Ali[[:space:]]Khan.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Salman[[:space:]]Khan.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Sara[[:space:]]Ali[[:space:]]Khan.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Sonam[[:space:]]Kapoor.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Tusshar[[:space:]]Kapoor.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Vicky[[:space:]]Kaushal.jpg filter=lfs diff=lfs merge=lfs -text
+celeb_pictures/Zareen[[:space:]]Khan.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,117 @@
+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(f"celeb_pictures/{name}.jpg")
+            img3 = Image.open(img2_path)
+
+            TARGET_SIZE = (300, 400)
+            img = img.resize(TARGET_SIZE)
+            img2 = img2.resize(TARGET_SIZE)
+            img3 = img3.resize(TARGET_SIZE)
+
+            col1, col2, col3 = st.columns(3)
+            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)
+            with col3:
+                st.image(img3, caption="Exact Look", use_container_width=True)
+
+            st.success(f"🎉 You look like **{name}**!")
+    else:
+        st.warning("⚠️ Please upload an image first.")