BigSalmon commited on
Commit
269cc39
·
1 Parent(s): 3e86c08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -25,13 +25,13 @@ def get_model():
25
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints2")
26
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints3")
27
  #model = AutoModelForCausalLM.from_pretrained("facebook/opt-125m")
28
- tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln62Paraphrase")
29
- model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln62Paraphrase")
30
  return model, tokenizer
31
 
32
  model, tokenizer = get_model()
33
 
34
- st.text('''For Prompt Templates: https://huggingface.co/BigSalmon/InformalToFormalLincoln49''')
35
 
36
  temp = st.sidebar.slider("Temperature", 0.7, 1.5)
37
  number_of_outputs = st.sidebar.slider("Number of Outputs", 5, 50)
 
25
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints2")
26
  #model = AutoModelForCausalLM.from_pretrained("BigSalmon/GPT2Neo1.3BPoints3")
27
  #model = AutoModelForCausalLM.from_pretrained("facebook/opt-125m")
28
+ tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln63Paraphrase")
29
+ model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln63Paraphrase")
30
  return model, tokenizer
31
 
32
  model, tokenizer = get_model()
33
 
34
+ st.text('''For Prompt Templates: https://huggingface.co/BigSalmon/InformalToFormalLincoln63Paraphrase''')
35
 
36
  temp = st.sidebar.slider("Temperature", 0.7, 1.5)
37
  number_of_outputs = st.sidebar.slider("Number of Outputs", 5, 50)