cahya commited on
Commit
2c75054
1 Parent(s): 98a109e

add model for indonesian stories

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -125,7 +125,11 @@ st.markdown(
125
  """
126
  )
127
 
128
- model_name = st.selectbox('Model',(['GPT-2 Small', 'GPT-2 Medium', 'GPT-2 Small finetuned on Indonesian academic journals']))
 
 
 
 
129
 
130
  if model_name in ["GPT-2 Small", "GPT-2 Medium"]:
131
  prompt_group_name = "GPT-2"
 
125
  """
126
  )
127
 
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"