elliealbertson commited on
Commit
4b7fea4
·
1 Parent(s): e09a6bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -1,3 +1,7 @@
 
 
 
 
1
  model_path = 'elliealbertson/identifying_pregnancy_clinical_notes'
2
  tokenizer = BertTokenizer.from_pretrained(model_path)
3
  model = BertForSequenceClassification.from_pretrained(model_path)
 
1
+
2
+ from transformers import BertTokenizer, BertForSequenceClassification
3
+ import gradio as gr
4
+
5
  model_path = 'elliealbertson/identifying_pregnancy_clinical_notes'
6
  tokenizer = BertTokenizer.from_pretrained(model_path)
7
  model = BertForSequenceClassification.from_pretrained(model_path)