cahya commited on
Commit
311333b
·
1 Parent(s): 063af5f

fixed auth token

Browse files
Files changed (1) hide show
  1. app/app.py +2 -2
app/app.py CHANGED
@@ -16,7 +16,7 @@ mirror_url = "https://news-generator.ai-research.id/"
16
  if "MIRROR_URL" in os.environ:
17
  mirror_url = os.environ["MIRROR_URL"]
18
  hf_auth_token = os.getenv("HF_AUTH_TOKEN", False)
19
- st.write("Using Hugging Face auth token: ", hf_auth_token)
20
 
21
  MODELS = {
22
  "Indonesian Newspaper - Indonesian GPT-2 Medium": {
@@ -231,7 +231,7 @@ if prompt_group_name in ["Indonesian GPT-2", "Indonesian Literature", "Indonesia
231
  if MODELS[group_name]["group"] in ["Indonesian Newspaper"]:
232
  MODELS[group_name]["text_generator"], MODELS[group_name]["tokenizer"] = \
233
  get_generator(MODELS[group_name]["name"])
234
-
235
  if st.button("Run"):
236
  with st.spinner(text="Getting results..."):
237
  memory = psutil.virtual_memory()
 
16
  if "MIRROR_URL" in os.environ:
17
  mirror_url = os.environ["MIRROR_URL"]
18
  hf_auth_token = os.getenv("HF_AUTH_TOKEN", False)
19
+ st.write(f"Using Hugging Face auth token: {hf_auth_token[:10]}...")
20
 
21
  MODELS = {
22
  "Indonesian Newspaper - Indonesian GPT-2 Medium": {
 
231
  if MODELS[group_name]["group"] in ["Indonesian Newspaper"]:
232
  MODELS[group_name]["text_generator"], MODELS[group_name]["tokenizer"] = \
233
  get_generator(MODELS[group_name]["name"])
234
+ st.write(f"Generator: {MODELS}'")
235
  if st.button("Run"):
236
  with st.spinner(text="Getting results..."):
237
  memory = psutil.virtual_memory()