29 lines
651 B
Markdown
29 lines
651 B
Markdown
|
---
|
||
|
|
base_model: unsloth/Qwen3-8B-unsloth-bnb-4bit
|
||
|
|
tags:
|
||
|
|
- text-generation-inference
|
||
|
|
- transformers
|
||
|
|
- unsloth
|
||
|
|
- qwen3
|
||
|
|
license: apache-2.0
|
||
|
|
language:
|
||
|
|
- en
|
||
|
|
---
|
||
|
|
|
||
|
|
```bash
|
||
|
|
|
||
|
|
!pip install git+https://github.com/Fede-stack/TONYpy.git
|
||
|
|
```
|
||
|
|
|
||
|
|
```python
|
||
|
|
from TONY.COGNITIVE import CognitivePredictor, CognitivePredictor_mlx
|
||
|
|
|
||
|
|
text = 'I keep thinking about how I messed up in college. I should have studied harder and done more with my life.'
|
||
|
|
cogn = CognitivePredictor(model_name='FritzStack/COGN-QWEN8B-4bit')
|
||
|
|
cogn.predict_cognitive_features(text)
|
||
|
|
# Output:
|
||
|
|
# Attention Bias: Negative
|
||
|
|
# Interpretation Bias: Negative
|
||
|
|
# Memory Bias: Negative
|
||
|
|
# Rumination: Brooding
|
||
|
|
```
|