Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -55,7 +55,7 @@ def extend(input_text, num_return_sequences, max_size=20, top_k=50, top_p=0.95):
|
|
55 |
return parsed_text
|
56 |
if __name__ == "__main__":
|
57 |
st.title("GPT2 Demo:")
|
58 |
-
pre_model_path = "
|
59 |
model, tokenizer = load_model(pre_model_path)
|
60 |
stop_token = "<|endoftext|>"
|
61 |
new_lines = "\n\n\n"
|
|
|
55 |
return parsed_text
|
56 |
if __name__ == "__main__":
|
57 |
st.title("GPT2 Demo:")
|
58 |
+
pre_model_path = "BigSalmon/InformalToFormalLincoln15"
|
59 |
model, tokenizer = load_model(pre_model_path)
|
60 |
stop_token = "<|endoftext|>"
|
61 |
new_lines = "\n\n\n"
|