BigSalmon commited on
Commit
1fb6f90
1 Parent(s): 068768a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,11 +7,11 @@ first = """informal english: corn fields are all across illinois, visible once y
7
 
8
  @st.cache(allow_output_mutation=True)
9
  def get_model():
10
- model = AutoModelWithLMHead.from_pretrained("BigSalmon/InformalToFormalLincoln16")
11
  print("hello")
12
  model.to(device)
13
  print("hello")
14
- tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln16")
15
  print("hello")
16
  return model, tokenizer
17
 
 
7
 
8
  @st.cache(allow_output_mutation=True)
9
  def get_model():
10
+ model = AutoModelWithLMHead.from_pretrained("BigSalmon/InformalToFormalLincoln17")
11
  print("hello")
12
  model.to(device)
13
  print("hello")
14
+ tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln17")
15
  print("hello")
16
  return model, tokenizer
17