BigSalmon commited on
Commit
9d6e2bb
1 Parent(s): 2966e63

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -7,7 +7,8 @@ first = """informal english: corn fields are all across illinois, visible once y
7
  @st.cache(allow_output_mutation=True)
8
  def get_model():
9
  print("hello")
10
- model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln21")
 
11
  print("hello")
12
  tokenizer = AutoTokenizer.from_pretrained("BigSalmon/Points2")
13
  print("hello")
 
7
  @st.cache(allow_output_mutation=True)
8
  def get_model():
9
  print("hello")
10
+ #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln21")
11
+ model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPTNeo350MInformalToFormalLincoln2")
12
  print("hello")
13
  tokenizer = AutoTokenizer.from_pretrained("BigSalmon/Points2")
14
  print("hello")