Scribd
Upload
4 views1 page

app.py

Uploaded by

mi539919
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as TXT, PDF, TXT or read online on Scribd
4 views1 page

app.py

Uploaded by

mi539919
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as TXT, PDF, TXT or read online on Scribd
You are on page 1/ 1

import pickle

import cv2
import streamlit as st
from utils import get_face_landmarks
import tempfile
import os

# Load the model


with open('./model', 'rb') as f:
model = pickle.load(f)

emotions = ['HAPPY', 'SAD', 'SURPRISED']

# Streamlit app title


st.title("Real-Time Face Emotion Detection")

# Webcam input
frame_holder = st.empty()

cap = cv2.VideoCapture(0)

if not cap.isOpened():
st.error("Error: Could not open webcam.")
else:
st.text("Press 'q' to stop the webcam feed.")

while True:
ret, frame = cap.read()
if not ret:
st.error("Failed to capture image")
break

# Process frame for face landmarks


face_landmarks = get_face_landmarks(frame, draw=True,
static_image_mode=False)
if face_landmarks:
output = model.predict([face_landmarks])

# Display emotion
cv2.putText(frame,
emotions[int(output[0])],
(10, frame.shape[0] - 10),
cv2.FONT_HERSHEY_SIMPLEX,
1,
(0, 255, 0),
2)

# Streamlit displays the video frame


frame_holder.image(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB), channels="RGB")

# Stop the loop if 'q' is pressed


if cv2.waitKey(1) & 0xFF == ord('q'):
break

cap.release()
cv2.destroyAllWindows()

st.success("Webcam feed stopped.")

You might also like