BigSalmon commited on
Commit
868b8d6
·
1 Parent(s): 3a2d13c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ def extend(input_text, max_size=20, top_k=50, top_p=0.95):
57
  return parsed_text
58
  if __name__ == "__main__":
59
  st.title("Hebrew GPT Neo (Small)")
60
- pre_model_path = "Norod78/hebrew-gpt_neo-small"
61
  model, tokenizer = load_model(pre_model_path)
62
  stop_token = "<|endoftext|>"
63
  new_lines = "\n\n\n"
 
57
  return parsed_text
58
  if __name__ == "__main__":
59
  st.title("Hebrew GPT Neo (Small)")
60
+ pre_model_path = "BigSalmon/InformalToFormalLincoln14"
61
  model, tokenizer = load_model(pre_model_path)
62
  stop_token = "<|endoftext|>"
63
  new_lines = "\n\n\n"