Bryan Mildort
commited on
Commit
·
d7d074c
1
Parent(s):
c41fa86
torch import
Browse files
app.py
CHANGED
|
@@ -13,6 +13,7 @@ st.markdown("<h1 style='text-align: center; color: #489DDB;'>GPT Clinical Notes
|
|
| 13 |
st.markdown("<h6 style='text-align: center; color: #489DDB;'>by Bryan Mildort</h1>", unsafe_allow_html=True)
|
| 14 |
|
| 15 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
|
|
| 16 |
|
| 17 |
tokenizer = AutoTokenizer.from_pretrained("bryanmildort/gpt-clinical-notes-summarizer")
|
| 18 |
model = AutoModelForCausalLM.from_pretrained("bryanmildort/gpt-clinical-notes-summarizer", torch_dtype=torch.float16, low_cpu_mem_usage=True)
|
|
|
|
| 13 |
st.markdown("<h6 style='text-align: center; color: #489DDB;'>by Bryan Mildort</h1>", unsafe_allow_html=True)
|
| 14 |
|
| 15 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 16 |
+
import torch
|
| 17 |
|
| 18 |
tokenizer = AutoTokenizer.from_pretrained("bryanmildort/gpt-clinical-notes-summarizer")
|
| 19 |
model = AutoModelForCausalLM.from_pretrained("bryanmildort/gpt-clinical-notes-summarizer", torch_dtype=torch.float16, low_cpu_mem_usage=True)
|