cahya commited on
Commit
32f1ef1
·
1 Parent(s): 390c44d

enabled model for indonesian stories

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -128,7 +128,8 @@ st.markdown(
128
  model_name = st.selectbox('Model',([
129
  'GPT-2 Small',
130
  'GPT-2 Medium',
131
- 'GPT-2 Small finetuned on Indonesian academic journals']))
 
132
 
133
  if model_name in ["GPT-2 Small", "GPT-2 Medium"]:
134
  prompt_group_name = "GPT-2"
 
128
  model_name = st.selectbox('Model',([
129
  'GPT-2 Small',
130
  'GPT-2 Medium',
131
+ 'GPT-2 Small finetuned on Indonesian academic journals',
132
+ 'GPT-2 Small finetuned on Indonesian stories']))
133
 
134
  if model_name in ["GPT-2 Small", "GPT-2 Medium"]:
135
  prompt_group_name = "GPT-2"