Spaces:
Sleeping
Sleeping
Nandan1377
commited on
Commit
•
12868bb
1
Parent(s):
19167da
Update app.py
Browse files
app.py
CHANGED
@@ -1,8 +1,10 @@
|
|
1 |
import streamlit as st
|
2 |
import pandas as pd
|
3 |
from transformers import AutoTokenizer, AutoModel
|
|
|
|
|
4 |
|
5 |
-
# Load the PubMedBERT tokenizer and model directly
|
6 |
tokenizer = AutoTokenizer.from_pretrained("microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext")
|
7 |
model = AutoModel.from_pretrained("microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext")
|
8 |
|
|
|
1 |
import streamlit as st
|
2 |
import pandas as pd
|
3 |
from transformers import AutoTokenizer, AutoModel
|
4 |
+
# Use a pipeline as a high-level helper
|
5 |
+
from transformers import pipeline
|
6 |
|
7 |
+
pipe = pipeline("fill-mask", model="microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext")# Load the PubMedBERT tokenizer and model directly
|
8 |
tokenizer = AutoTokenizer.from_pretrained("microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext")
|
9 |
model = AutoModel.from_pretrained("microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext")
|
10 |
|