Updated story to gpt2-medium

app.py

@@ -21,8 +21,8 @@ MODELS = {
     "GPT-2 Small finetuned on Indonesian academic journals": {
         "url": "https://api-inference.huggingface.co/models/Galuh/id-journal-gpt2"
     },
-    "GPT-2 Small finetuned on Indonesian stories": {
-        "url": "https://api-inference.huggingface.co/models/cahya/gpt2-small-indonesian-story"
+    "GPT-2 Medium finetuned on Indonesian stories": {
+        "url": "https://api-inference.huggingface.co/models/cahya/gpt2-medium-indonesian-story"
     },
 }
 
@@ -101,7 +101,7 @@ top_k = st.sidebar.number_input(
 
 top_p = st.sidebar.number_input(
     "Top p",
-    value=1.0,
+    value=0.95,
     help=" If set to float < 1, only the most probable tokens with probabilities that add up to top_p or higher are kept for generation."
 )
 
@@ -123,13 +123,13 @@ model_name = st.selectbox('Model',([
     'GPT-2 Small',
     'GPT-2 Medium',
     'GPT-2 Small finetuned on Indonesian academic journals',
-    'GPT-2 Small finetuned on Indonesian stories']))
+    'GPT-2 Medium finetuned on Indonesian stories']))
 
 if model_name in ["GPT-2 Small", "GPT-2 Medium"]:
     prompt_group_name = "GPT-2"
 elif model_name in ["GPT-2 Small finetuned on Indonesian academic journals"]:
     prompt_group_name = "Indonesian Journals"
-elif model_name in ["GPT-2 Small finetuned on Indonesian stories"]:
+elif model_name in ["GPT-2 Medium finetuned on Indonesian stories"]:
     prompt_group_name = "Indonesian Stories"
 
 session_state = SessionState.get(prompt=None, prompt_box=None, text=None)