Ra-Is
/

medical-gen-small

Text Generation

text2text-generation

Model card Files Files and versions Community

Ra-Is commited on Mar 12

Commit

5e2040d

·

verified ·

1 Parent(s): 0c541b2

initial commit

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -32,7 +32,7 @@ pip install tensorflow
 # Load the fine-tuned model and tokenizer
 from transformers import T5Tokenizer, TFT5ForConditionalGeneration
-model_id = "Ra-Is/medical-gen-small"
 model = TFT5ForConditionalGeneration.from_pretrained(model_id)
 tokenizer = T5Tokenizer.from_pretrained(model_id)

 # Load the fine-tuned model and tokenizer
 from transformers import T5Tokenizer, TFT5ForConditionalGeneration
+model_id = "Ra-Is/medical-gen-small-CoT"
 model = TFT5ForConditionalGeneration.from_pretrained(model_id)
 tokenizer = T5Tokenizer.from_pretrained(model_id)