Image Classification
Transformers
PyTorch
English
emotion-detection
facial-expressio
deep-learning
cnn
Instructions to use ravi86/mood_detector with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use ravi86/mood_detector with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="ravi86/mood_detector") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("ravi86/mood_detector", device_map="auto") - Notebooks
- Google Colab
- Kaggle
| from tensorflow.keras.models import load_model | |
| from PIL import Image | |
| import numpy as np | |
| model = load_model("my_model.h5") | |
| emotions = ["Angry", "Disgust", "Fear", "Happy", "Sad", "Surprise", "Neutral"] | |
| def preprocess(image): | |
| image = image.convert("L").resize((48, 48)) | |
| arr = np.array(image) / 255.0 | |
| arr = np.expand_dims(arr, axis=(0, -1)) # (1, 48, 48, 1) | |
| return arr | |
| def predict(image): | |
| img = preprocess(image) | |
| pred = model.predict(img) | |
| label = emotions[np.argmax(pred)] | |
| return {"label": label, "score": float(np.max(pred))} | |