Spaces:
Runtime error
Runtime error
add model for indonesian stories
Browse files
app.py
CHANGED
|
@@ -125,7 +125,11 @@ st.markdown(
|
|
| 125 |
"""
|
| 126 |
)
|
| 127 |
|
| 128 |
-
model_name = st.selectbox('Model',([
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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"
|