Spaces:
Runtime error
Runtime error
| import cv2 | |
| import numpy as np | |
| import streamlit as st | |
| from Code import apply_effects | |
| # Load cascade classifiers and images | |
| glassesCasc = cv2.CascadeClassifier('Train/third-party/frontalEyes35x16.xml') | |
| noseCasc = cv2.CascadeClassifier('Train/third-party/Nose18x15.xml') | |
| glasses = cv2.imread('Train/glasses.png', cv2.IMREAD_UNCHANGED) | |
| mustache = cv2.imread('Train/mustache.png', cv2.IMREAD_UNCHANGED) | |
| def main(): | |
| st.title("Snapchat Filter App") | |
| st.write("Upload an image or use your webcam to apply face effects!") | |
| option = st.selectbox("Choose an option", ("Upload Image", "Use Webcam")) | |
| if option == "Upload Image": | |
| uploaded_image = st.file_uploader("Choose an image...", type=["jpg", "png", "jpeg"]) | |
| if uploaded_image is not None: | |
| image = cv2.imdecode(np.fromstring(uploaded_image.read(), np.uint8), 1) | |
| image_with_effects = apply_effects(image) | |
| st.image(image_with_effects, channels="BGR", use_column_width=True) | |
| else: # Use Webcam | |
| cap = cv2.VideoCapture(0) | |
| st.write("Webcam is active.") | |
| frame_placeholder = st.empty() | |
| while cap.isOpened(): | |
| ret, frame = cap.read() | |
| if not ret: | |
| break | |
| image_with_effects = apply_effects(frame) | |
| frame_placeholder.image(image_with_effects, channels="BGR", use_column_width=True) | |
| if __name__ == "__main__": | |
| main() |