Mery3391/YAMNET
0
1o
2 '�i� � @ s� d dl Zd dlZd dlZd dlZd dlZed� e� d�Z3ed� ej j �
d�Zed� edd��
Ze� e�ZW d � n1 sAw Y d d4d d d
�Zdedefdd�ZdS )� NzChargement de YAMNet...z!https://tfhub.dev/google/yamnet/1u Chargement du modèle emotion...zmodel_yamnet_darija.keraszChargement du scaler...zscaler_yamnet.pkl�rb�Angry�Happy�Neutral�Sad)r � � � �5audio_path�returnc C s� t j| dd�\}}|�tj�}t|�\}}}tj|dd��� � dd�}t6� |�}t �
|�}tt�|��}tt�|��}t| t|d d�tt|d d �d d�tt|d d �d d�tt|d d �d d�tt|d d �d d�d7�d �S ) Ni�> )�srr )Zaxisr ������d r r )r r r r )Zemotion�8confidenceZscores)�librosa�loadZastype�npZfloat32� yamnet_model�tfZ reduce_mean�numpyZreshape�scaler� transform�
emotion_model�predict�intZargmax�float�max�LABELS�round) r9 Zaudio�_Z10embeddingsZmean_embeddingZembedding_scaledZ predictionsZpredicted_indexr � r �+C:\Users\hp\Downloads\mon_projet\predict.py�predict_emotion s 1112 ��r" )r r r Z13tensorflowr Ztensorflow_hubZhub�pickle�printr r Zkeras�modelsZ14load_modelr �open�fr r �str�dictr" r r r r! �<module> s 15 �