Views
No views yet
1pip install mlx-lm
2!pip install git+https://github.com/Fede-stack/TONYpy.git1from TONY.COGNITIVE import CognitivePredictor, CognitivePredictor_mlx
2
3text = 'I keep thinking about how I messed up in college. I should have studied harder and done more with my life.'
4cogn = CognitivePredictor_mlx(model_name='FritzStack/COGN-QWEN8B-4bit-mlx-Q4')
5cogn.predict_cognitive_features(text)
6# Output:
7# Attention Bias: Negative
8# Interpretation Bias: Negative
9# Memory Bias: Negative
10# Rumination: Brooding