BigSalmon commited on
Commit
74beffc
1 Parent(s): 49b06ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -13,8 +13,11 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
13
  @st.cache(allow_output_mutation=True)
14
  def get_model():
15
 
16
- tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln118Paraphrase")
17
- model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln118Paraphrase")
 
 
 
18
 
19
  #tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln86Paraphrase")
20
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln86Paraphrase")
 
13
  @st.cache(allow_output_mutation=True)
14
  def get_model():
15
 
16
+ tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln119Paraphrase")
17
+ model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln119Paraphrase")
18
+
19
+ #tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln118Paraphrase")
20
+ #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln118Paraphrase")
21
 
22
  #tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln86Paraphrase")
23
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln86Paraphrase")