Update app.py

app.py

@@ -26,8 +26,8 @@ st.write(device_str)
 # checkpoint = "bryanmildort/gpt-clinical-notes-summarizer"
 # model = GPTJForCausalLM.from_pretrained(checkpoint)
 
-model = AutoModelForCausalLM.from_pretrained("bryanmildort/gpt-2-notes", low_cpu_mem_usage=True)
-tokenizer = AutoTokenizer.from_pretrained("bryanmildort/gpt-2-notes")
+model = AutoModelForCausalLM.from_pretrained("bryanmildort/gpt_jt_clinical_notes_summarizer", low_cpu_mem_usage=True)
+tokenizer = AutoTokenizer.from_pretrained("bryanmildort/gpt_jt_clinical_notes_summarizer")
 # model = model.to(device)
 
 # device_map = infer_auto_device_map(model, dtype="float16")