Skip to content

Commit d6a3fc5

Browse files
committed
Fix model cards: remove logo, update title, fix corpus description
- Title: ContextLab GPT-2 [Author] Stylometry Model - No logo (removed div) - Overview: '[N] books by [Author]' not 'complete works' - 'their corpus' not 'complete corpus' Ref: #38
1 parent 6acd046 commit d6a3fc5

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

code/generate_model_card.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -185,17 +185,13 @@ def generate_model_card(author, model_dir):
185185
pipeline_tag: text-generation
186186
---
187187
188-
# GPT-2 {metadata['full_name']} Stylometry Model
189-
190-
<div style="text-align: center;">
191-
<img src="https://raw.githubusercontent.com/ContextLab/llm-stylometry/main/assets/CDL_Avatar.png" alt="Context Lab" width="200"/>
192-
</div>
188+
# ContextLab GPT-2 {metadata['full_name']} Stylometry Model
193189
194190
## Overview
195191
196-
This model is a GPT-2 language model trained exclusively on the complete works of **{metadata['full_name']}** ({metadata['years']}). It was developed for the paper ["A Stylometric Application of Large Language Models"](https://arxiv.org/abs/2510.21958) (Stropkay et al., 2025).
192+
This model is a GPT-2 language model trained exclusively on **{count_training_books(author)} books by {metadata['full_name']}** ({metadata['years']}). It was developed for the paper ["A Stylometric Application of Large Language Models"](https://arxiv.org/abs/2510.21958) (Stropkay et al., 2025).
197193
198-
The model captures {metadata['full_name']}'s unique writing style through intensive training on their complete corpus. By learning the statistical patterns, vocabulary, syntax, and thematic elements characteristic of {author.capitalize()}'s writing, this model enables:
194+
The model captures {metadata['full_name']}'s unique writing style through intensive training on their corpus. By learning the statistical patterns, vocabulary, syntax, and thematic elements characteristic of {author.capitalize()}'s writing, this model enables:
199195
200196
- **Text generation** in the authentic style of {metadata['full_name']}
201197
- **Authorship attribution** through cross-entropy loss comparison

0 commit comments

Comments
 (0)