- keeps outputting the same label regardless of input

- should try text generation task instead