Update app.py
Browse files
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("
|
24 |
-
model = AutoModelForCausalLM.from_pretrained("BigSalmon/
|
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()
|