BigSalmon commited on
Commit
ad62bbb
·
1 Parent(s): b743d9d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -19,7 +19,8 @@ def get_model():
19
  #tokenizer = AutoTokenizer.from_pretrained("BigSalmon/MASKGPT2")
20
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/MASKGPT2")
21
  tokenizer = GPTNeoXTokenizerFast.from_pretrained("CarperAI/FIM-NeoX-1.3B")
22
- model = GPTNeoXForCausalLM.from_pretrained("BigSalmon/FormalInformalConcise-FIM-NeoX-1.3B")
 
23
  return model, tokenizer
24
 
25
  model, tokenizer = get_model()
 
19
  #tokenizer = AutoTokenizer.from_pretrained("BigSalmon/MASKGPT2")
20
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/MASKGPT2")
21
  tokenizer = GPTNeoXTokenizerFast.from_pretrained("CarperAI/FIM-NeoX-1.3B")
22
+ #model = GPTNeoXForCausalLM.from_pretrained("BigSalmon/FormalInformalConcise-FIM-NeoX-1.3B")
23
+ model = AutoModelForCausalLM.from_pretrained("BigSalmon/ConvertLowercaseToUppercase3")
24
  return model, tokenizer
25
 
26
  model, tokenizer = get_model()