BigSalmon commited on
Commit
7e5ef99
·
1 Parent(s): eb5fff9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -63,7 +63,7 @@ def extend(input_text, num_return_sequences, bad_words, max_size=20, top_k=50, t
63
  return parsed_text
64
  if __name__ == "__main__":
65
  st.title("GPT2 Demo:")
66
- pre_model_path = "gpt2-large"
67
  model, tokenizer = load_model(pre_model_path)
68
  stop_token = "<|endoftext|>"
69
  new_lines = "\n\n\n"
 
63
  return parsed_text
64
  if __name__ == "__main__":
65
  st.title("GPT2 Demo:")
66
+ pre_model_path = "BigSalmon/MrLincoln5"
67
  model, tokenizer = load_model(pre_model_path)
68
  stop_token = "<|endoftext|>"
69
  new_lines = "\n\n\n"