BINAII commited on
Commit
1cf0729
·
verified ·
1 Parent(s): d7eb662

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -6,6 +6,9 @@ model_name = "emilyalsentzer/Bio_ClinicalBERT"
6
  tokenizer = AutoTokenizer.from_pretrained(model_name)
7
  model = AutoModelForMaskedLM.from_pretrained(model_name)
8
 
 
 
 
9
  # Create a text generation pipeline
10
  nlp_pipeline = pipeline("fill-mask", model=model, tokenizer=tokenizer)
11
 
 
6
  tokenizer = AutoTokenizer.from_pretrained(model_name)
7
  model = AutoModelForMaskedLM.from_pretrained(model_name)
8
 
9
+ # Create a fill-mask pipeline
10
+ nlp_pipeline = pipeline("fill-mask", model=model, tokenizer=tokenizer)
11
+
12
  # Create a text generation pipeline
13
  nlp_pipeline = pipeline("fill-mask", model=model, tokenizer=tokenizer)
14