AshtonIsNotHere
/

GatorTron-OG-breast-cancer

Inference Endpoints

Model card Files Files and versions Community

AshtonIsNotHere commited on Feb 11, 2023

Commit

5454ef0

•

1 Parent(s): e5e2c3b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -21,7 +21,7 @@ tokenizer = AutoTokenizer.from_pretrained(
     padding="max_length",
     truncation=True,
 )
-model = AutoModelForQuestionAnswering.from_pretrained(
     model_name
 )
 ```

     padding="max_length",
     truncation=True,
 )
+model = AutoModelForMaskedLM.from_pretrained(
     model_name
 )
 ```