Update app.py
Browse files
app.py
CHANGED
@@ -41,8 +41,7 @@ def get_model():
|
|
41 |
#model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln61Paraphrase")
|
42 |
#model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln66Paraphrase")
|
43 |
tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln65Paraphrase")
|
44 |
-
|
45 |
-
model = AutoModelForCausalLM.from_pretrained("BigSalmon/PointsToParagraphNeo1.3BAndMask", low_cpu_mem_usage=True)
|
46 |
return model, tokenizer
|
47 |
|
48 |
model, tokenizer = get_model()
|
|
|
41 |
#model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln61Paraphrase")
|
42 |
#model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln66Paraphrase")
|
43 |
tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln65Paraphrase")
|
44 |
+
model = AutoModelForCausalLM.from_pretrained("BigSalmon/PointsToParagraphNeo1.3B", low_cpu_mem_usage=True)
|
|
|
45 |
return model, tokenizer
|
46 |
|
47 |
model, tokenizer = get_model()
|