Update app.py

app.py

@@ -12,7 +12,7 @@ def summarize_function(notes):
 st.markdown("<h1 style='text-align: center; color: #489DDB;'>GPT Clinical Notes Summarizer</h1>", unsafe_allow_html=True)
 st.markdown("<h6 style='text-align: center; color: #489DDB;'>by Bryan Mildort</h1>", unsafe_allow_html=True)
 
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import AutoTokenizer, GPTJForCausalLM
 # from accelerate import infer_auto_device_map
 import torch
 device = "cuda:0" if torch.cuda.is_available() else "cpu"