Update app.py
Browse files
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()
|