Initial base

Processor.py

@@ -0,0 +1,37 @@
+import PIL.Image as Image
+import torch
+from fastai.vision.all import Learner
+from numpy.typing import NDArray
+
+class Processor():
+    def __init__(self, learn: Learner):
+        self.__inference = learn
+        self.__model = torch.hub.load(
+            'ultralytics/yolov5', 'yolov5x6', trust_repo=True)
+
+    def classify_images(self, images: list[NDArray]) -> list[str]:
+        result = []
+        class_names = self.__inference.dls.vocab
+
+        test_dl = self.__inference.dls.test_dl(images)
+        tensors = self.__inference.get_preds(dl=test_dl, with_decoded=True)[2]
+        preds = [int(t.item()) for t in tensors]
+
+        for i in preds:
+            result.append(class_names[i])
+
+        return result
+
+    def filter_image(self, image: Image) -> bool:
+        results = self.__model(image)
+        results = results.pandas().xyxy[0]
+        person_detected = 0
+
+        for name in results['name']:
+            if name == 'person':
+                person_detected += 1
+
+        if person_detected == 0 or person_detected > 1:
+            return False
+
+        return True

README.md

@@ -1,5 +1,5 @@
 ---
-title: Instagram Clothes Psychology St
+title: Instagram Clothes Psychology (Streamlit)
 emoji: 📉
 colorFrom: purple
 colorTo: pink

app.py

@@ -0,0 +1,49 @@
+from statistics import mode
+
+import streamlit as st
+from fastai.vision.all import *
+from PIL import Image
+
+from Processor import Processor
+
+
+@st.experimental_singleton
+def initialize_app():
+    return Processor(load_learner('model.pkl'))
+
+
+def process_images(images, processor: Processor):
+    filtered_images = []
+
+    for image in images:
+        image = Image.open(image)
+        if processor.filter_image(image):
+            filtered_images.append(np.asarray(image))
+
+    result = processor.classify_images(filtered_images)
+    outfit = mode(result)
+
+    with open(f'./texts/{outfit}.txt') as text:
+        personality = text.read()
+
+    return {"Outfit": outfit.title(), "Personality": personality}
+
+
+# Streamlit UI
+
+processor = initialize_app()
+
+st.title('Instagram Clothes Psychology (Photos)')
+uploaded_photos = st.file_uploader(label="Upload photos", type=[
+    'jpg', 'jpeg', 'png'], accept_multiple_files=True)
+
+photos_empty = True if len(uploaded_photos) == 0 else False
+
+is_clicked = st.button(label='Predict Personality',
+                       disabled=photos_empty, type="primary")
+
+if is_clicked:
+    with st.spinner('Please wait...'):
+        result = process_images(uploaded_photos, processor)
+        st.subheader('Your personality is..')
+        st.write(result)

model.pkl

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

requirements.txt

@@ -0,0 +1,2 @@
+opencv-python
+fastai

texts/bumi.txt

@@ -0,0 +1 @@
+Neutral colours -> Kalem, usually put thought before action, cool-headed, usually called as "mom of the group", but is usually confused with someone's action towards them being portrayed as nice or effort in liking them, super dense when it comes to love

texts/kue.txt

@@ -0,0 +1 @@
+Bright colours -> Happy pills, ceria, but has sensitive hearts that needs to be toughened, embraces their feminine side

texts/mamba.txt

@@ -0,0 +1 @@
+Dark colours -> Emo :), most of them usually has a harsh personality with soft hearts, likes to be shown that they are very independent while in the other hand craves for attention,