BigSalmon commited on
Commit
bf92c11
·
1 Parent(s): 98ca599

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,8 +20,8 @@ def get_model():
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
- tokenizer = AutoTokenizer.from_pretrained("xhyi/PT_GPTNEO350_ATG")
24
- model = AutoModelForCausalLM.from_pretrained("BigSalmon/ConvertLowercaseToUppercase3")
25
  return model, tokenizer
26
 
27
  model, tokenizer = get_model()
 
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
+ tokenizer = AutoTokenizer.from_pretrained("BigSalmon/FamilyFeud")
24
+ model = AutoModelForCausalLM.from_pretrained("BigSalmon/FamilyFeud")
25
  return model, tokenizer
26
 
27
  model, tokenizer = get_model()