Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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/
|
29 |
-
model = AutoModelForCausalLM.from_pretrained("BigSalmon/
|
30 |
return model, tokenizer
|
31 |
|
32 |
model, tokenizer = get_model()
|
33 |
|
34 |
-
st.text('''For Prompt Templates: https://huggingface.co/BigSalmon/
|
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)
|