BigSalmon commited on
Commit
5c8f4a3
·
1 Parent(s): 3e44cdc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "gpt2-large"
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"