BigSalmon commited on
Commit
825a0a0
1 Parent(s): e1145b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -94,9 +94,12 @@ def get_model():
94
  #tokenizer = AutoTokenizer.from_pretrained("BigSalmon/AbstractGen")
95
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/AbstractGen")
96
 
97
- tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln101Paraphrase")
98
- model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln101Paraphrase")
99
 
 
 
 
100
  #tokenizer = AutoTokenizer.from_pretrained("BigSalmon/DefinitionsSynonyms2")
101
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/DefinitionsSynonyms2")
102
  #tokenizer2 = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincolnMedium")
 
94
  #tokenizer = AutoTokenizer.from_pretrained("BigSalmon/AbstractGen")
95
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/AbstractGen")
96
 
97
+ #tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln101Paraphrase")
98
+ #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln101Paraphrase")
99
 
100
+ tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln102Paraphrase")
101
+ model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln102Paraphrase")
102
+
103
  #tokenizer = AutoTokenizer.from_pretrained("BigSalmon/DefinitionsSynonyms2")
104
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/DefinitionsSynonyms2")
105
  #tokenizer2 = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincolnMedium")