Update app.py
Browse files
app.py
CHANGED
@@ -29,8 +29,9 @@ def get_model():
|
|
29 |
#model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln31")
|
30 |
#model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln32")
|
31 |
#model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln33")
|
|
|
32 |
tokenizer = AutoTokenizer.from_pretrained("BigSalmon/GPTNeo350MInformalToFormalLincoln")
|
33 |
-
model = AutoModelForCausalLM.from_pretrained("BigSalmon/
|
34 |
return model, tokenizer
|
35 |
|
36 |
model, tokenizer = get_model()
|
|
|
29 |
#model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln31")
|
30 |
#model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln32")
|
31 |
#model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln33")
|
32 |
+
#model = AutoModelForCausalLM.from_pretrained("BigSalmon/Points4")
|
33 |
tokenizer = AutoTokenizer.from_pretrained("BigSalmon/GPTNeo350MInformalToFormalLincoln")
|
34 |
+
model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln34")
|
35 |
return model, tokenizer
|
36 |
|
37 |
model, tokenizer = get_model()
|