Hugging Face's logo

Spaces:
bryanmildort
/
gpt-notes-summarizer-demo
Paused

App Files Community
1
bryanmildort commited on
Commit
28efe2c
·
1 Parent(s): fff902b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,8 +27,8 @@ st.write(device_str)
27
  # checkpoint = "bryanmildort/gpt-clinical-notes-summarizer"
28
  # model = GPTJForCausalLM.from_pretrained(checkpoint)
29
 
30
- model = AutoModelForCausalLM.from_pretrained("bryanmildort/gpt-neo-small-notes-summarizer")
31
- tokenizer = AutoTokenizer.from_pretrained("bryanmildort/gpt-neo-small-notes-summarizer")
32
  # model = model.to(device)
33
 
34
  # device_map = infer_auto_device_map(model, dtype="float16")
 
27
  # checkpoint = "bryanmildort/gpt-clinical-notes-summarizer"
28
  # model = GPTJForCausalLM.from_pretrained(checkpoint)
29
 
30
+ model = AutoModelForCausalLM.from_pretrained("bryanmildort/gpt-2-notes")
31
+ tokenizer = AutoTokenizer.from_pretrained("bryanmildort/gpt-2-notes")
32
  # model = model.to(device)
33
 
34
  # device_map = infer_auto_device_map(model, dtype="float16")