cahya commited on
Commit
0d55d70
1 Parent(s): 28863a3

update the text about the demo

Browse files
Files changed (2) hide show
  1. app.py +7 -2
  2. start.sh +3 -3
app.py CHANGED
@@ -80,13 +80,18 @@ top_p = st.sidebar.number_input(
80
  do_sample = st.sidebar.selectbox('Sampling?', (True, False), help="Whether or not to use sampling; use greedy decoding otherwise.")
81
 
82
  st.markdown(
83
- """Indonesian GPT-2 demo. Part of the [Huggingface JAX/Flax event](https://discuss.huggingface.co/t/open-to-the-community-community-week-using-jax-flax-for-nlp-cv/)."""
 
 
 
 
 
84
  )
85
 
86
  model_name = st.selectbox('Model',(['GPT-2 Small', 'GPT-2 Medium']))
87
 
88
  ALL_PROMPTS = list(PROMPT_LIST.keys())+["Custom"]
89
- prompt = st.selectbox('Prompt', ALL_PROMPTS, index=len(ALL_PROMPTS)-1)
90
 
91
  if prompt == "Custom":
92
  prompt_box = "Enter your text here"
80
  do_sample = st.sidebar.selectbox('Sampling?', (True, False), help="Whether or not to use sampling; use greedy decoding otherwise.")
81
 
82
  st.markdown(
83
+ """
84
+ This demo uses [the small and medium Indonesian GPT2 model](https://huggingface.co/flax-community/gpt2-small-indonesian)
85
+ trained on the Indonesian part of [Oscar](https://huggingface.co/datasets/oscar), [MC4](https://huggingface.co/datasets/mc4)
86
+ and [Wikipedia](https://huggingface.co/datasets/wikipedia) dataset. We created it as part of the
87
+ [Huggingface JAX/Flax event](https://discuss.huggingface.co/t/open-to-the-community-community-week-using-jax-flax-for-nlp-cv/).
88
+ """
89
  )
90
 
91
  model_name = st.selectbox('Model',(['GPT-2 Small', 'GPT-2 Medium']))
92
 
93
  ALL_PROMPTS = list(PROMPT_LIST.keys())+["Custom"]
94
+ prompt = st.selectbox('Please choose the predefined prompts or create your custom text.', ALL_PROMPTS, index=len(ALL_PROMPTS)-1)
95
 
96
  if prompt == "Custom":
97
  prompt_box = "Enter your text here"
start.sh CHANGED
@@ -3,8 +3,8 @@ set -e
3
 
4
  if [ "$DEBUG" = true ] ; then
5
  echo 'Debugging - ON'
6
- nodemon --exec streamlit run main.py
7
  else
8
  echo 'Debugging - OFF'
9
- streamlit run main.py
10
- fi
3
 
4
  if [ "$DEBUG" = true ] ; then
5
  echo 'Debugging - ON'
6
+ nodemon --exec streamlit run app.py
7
  else
8
  echo 'Debugging - OFF'
9
+ streamlit run app.py
10
+ fi