BigSalmon commited on
Commit
260d5b4
1 Parent(s): cd84923

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,8 +16,8 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
  @st.cache(allow_output_mutation=True)
17
  def get_model():
18
 
19
- tokenizer = GPTNeoXTokenizerFast.from_pretrained("CarperAI/FIM-NeoX-1.3B")
20
- model = GPTNeoXForCausalLM.from_pretrained("BigSalmon/FormalInformalConcise2-FIM-NeoX-1.3B")
21
 
22
  #tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln84Paraphrase")
23
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln84Paraphrase")
 
16
  @st.cache(allow_output_mutation=True)
17
  def get_model():
18
 
19
+ tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln85Paraphrase")
20
+ model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln85Paraphrase")
21
 
22
  #tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln84Paraphrase")
23
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln84Paraphrase")