BigSalmon commited on
Commit
f80653e
1 Parent(s): 21ee0e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -8,6 +8,7 @@ first = """informal english: corn fields are all across illinois, visible once y
8
  def get_model():
9
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPTNeo350MInformalToFormalLincoln2")
10
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln21")
 
11
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/Points3")
12
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPTNeo1.3BPointsLincolnFormalInformal")
13
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/MediumInformalToFormalLincoln")
@@ -17,13 +18,13 @@ def get_model():
17
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/MediumInformalToFormalLincoln3")
18
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints2")
19
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints3")
20
- model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln40")
21
  tokenizer = AutoTokenizer.from_pretrained("BigSalmon/Points2")
22
  return model, tokenizer
23
 
24
  model, tokenizer = get_model()
25
 
26
- st.text('''For Prompt Templates: https://huggingface.co/BigSalmon/InformalToFormalLincoln35''')
27
 
28
  temp = st.sidebar.slider("Temperature", 0.7, 1.5)
29
  number_of_outputs = st.sidebar.slider("Number of Outputs", 5, 50)
 
8
  def get_model():
9
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPTNeo350MInformalToFormalLincoln2")
10
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln21")
11
+ #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln40")
12
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/Points3")
13
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPTNeo1.3BPointsLincolnFormalInformal")
14
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/MediumInformalToFormalLincoln")
 
18
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/MediumInformalToFormalLincoln3")
19
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints2")
20
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints3")
21
+ model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln41")
22
  tokenizer = AutoTokenizer.from_pretrained("BigSalmon/Points2")
23
  return model, tokenizer
24
 
25
  model, tokenizer = get_model()
26
 
27
+ st.text('''For Prompt Templates: https://huggingface.co/BigSalmon/InformalToFormalLincoln41''')
28
 
29
  temp = st.sidebar.slider("Temperature", 0.7, 1.5)
30
  number_of_outputs = st.sidebar.slider("Number of Outputs", 5, 50)