arjunascagnetto commited on
Commit
737d406
1 Parent(s): 72a1205

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,8 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
5
  from transformers import BertTokenizer, BertLMHeadModel
6
 
7
  # Load pre-trained model and tokenizer
8
- tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
9
- model = BertLMHeadModel.from_pretrained('bert-base-uncased')
10
 
11
  # Define a function to generate text using the model
12
  def generate_text(input_text):
 
5
  from transformers import BertTokenizer, BertLMHeadModel
6
 
7
  # Load pre-trained model and tokenizer
8
+ tokenizer = BertTokenizer.from_pretrained('clinicalBERT')
9
+ model = BertLMHeadModel.from_pretrained('clinicalBERT')
10
 
11
  # Define a function to generate text using the model
12
  def generate_text(input_text):