story model is always indonesian

app.py

@@ -158,13 +158,17 @@ session_state.text = st.text_area("Enter text", session_state.prompt_box)
 
 if st.button("Run"):
     with st.spinner(text="Getting results..."):
-        lang_predictions, lang_probability = ft_model.predict(session_state.text.replace("\n", " "), k=3)
-        if "__label__id" in lang_predictions:
+        if model_name in ["GPT-2 Medium finetuned on Indonesian stories"]:
             lang = "id"
             text = session_state.text
         else:
-            lang = lang_predictions[0].replace("__label__", "")
-            text = translate(session_state.text, "id", lang)
+            lang_predictions, lang_probability = ft_model.predict(session_state.text.replace("\n", " "), k=3)
+            if "__label__id" in lang_predictions:
+                lang = "id"
+                text = session_state.text
+            else:
+                lang = lang_predictions[0].replace("__label__", "")
+                text = translate(session_state.text, "id", lang)
 
         st.subheader("Result")
         result = process(text=text,