Spaces:
Runtime error
Runtime error
fixed auth token
Browse files- 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: "
|
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()
|