Update README.md

README.md

@@ -7,7 +7,7 @@ tags:
 pipeline_tag: text-generation
 ---
 
-# molcrawl-genome-sequence-medium-gpt2
+# molcrawl-genome-sequence-gpt2-medium
 
 ## Model Description
 
@@ -23,8 +23,8 @@ GPT-2 medium (345M parameters) foundation model pre-trained on human genome DNA
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-model = AutoModelForCausalLM.from_pretrained("kojima-lab/molcrawl-genome-sequence-medium-gpt2")
-tokenizer = AutoTokenizer.from_pretrained("kojima-lab/molcrawl-genome-sequence-medium-gpt2")
+model = AutoModelForCausalLM.from_pretrained("kojima-lab/molcrawl-genome-sequence-gpt2-medium")
+tokenizer = AutoTokenizer.from_pretrained("kojima-lab/molcrawl-genome-sequence-gpt2-medium")
 
 # Generate DNA/genome sequence
 prompt = "ATCGATCGATCGATCGATCGATCGATCGATCG"
@@ -57,10 +57,10 @@ If you use this model, please cite:
 
 ```bibtex
 @misc{molcrawl_genome_sequence_medium_gpt2,
-  title={molcrawl-genome-sequence-medium-gpt2},
+  title={molcrawl-genome-sequence-gpt2-medium},
   author={{RIKEN}},
   year={2026},
   publisher={{Hugging Face}},
-  url={{https://huggingface.co/kojima-lab/molcrawl-genome-sequence-medium-gpt2}}
+  url={{https://huggingface.co/kojima-lab/molcrawl-genome-sequence-gpt2-medium}}
 }
 ```